py_api.rst 2.35 KB
Newer Older
1 2
.. vim: set fileencoding=utf-8 :

Amir MOHAMMADI's avatar
Amir MOHAMMADI committed
3
.. _bob.learn.tensorflow.py_api:
Tiago de Freitas Pereira's avatar
Tiago de Freitas Pereira committed
4

5 6 7 8
============
 Python API
============

Tiago de Freitas Pereira's avatar
Tiago de Freitas Pereira committed
9

10 11
Models
======
Tiago de Freitas Pereira's avatar
Tiago de Freitas Pereira committed
12

13 14 15 16 17 18
.. autosummary::
    bob.learn.tensorflow.models.AlexNet_simplified
    bob.learn.tensorflow.models.DeepPixBiS
    bob.learn.tensorflow.models.DenseNet
    bob.learn.tensorflow.models.densenet161
    bob.learn.tensorflow.models.MineModel
Tiago de Freitas Pereira's avatar
Tiago de Freitas Pereira committed
19

Amir MOHAMMADI's avatar
Amir MOHAMMADI committed
20 21
Data
====
Tiago de Freitas Pereira's avatar
Tiago de Freitas Pereira committed
22

Amir MOHAMMADI's avatar
Amir MOHAMMADI committed
23
.. autosummary::
24 25 26
    bob.learn.tensorflow.data.dataset_using_generator
    bob.learn.tensorflow.data.dataset_to_tfrecord
    bob.learn.tensorflow.data.dataset_from_tfrecord
27 28 29 30 31 32


Losses
======

.. autosummary::
33 34
    bob.learn.tensorflow.losses.CenterLossLayer
    bob.learn.tensorflow.losses.CenterLoss
35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60
    bob.learn.tensorflow.losses.PixelwiseBinaryCrossentropy
    bob.learn.tensorflow.losses.balanced_sigmoid_cross_entropy_loss_weights
    bob.learn.tensorflow.losses.balanced_softmax_cross_entropy_loss_weights


Image Utilities
===============

.. autosummary::
    bob.learn.tensorflow.utils.image.to_channels_last
    bob.learn.tensorflow.utils.image.to_channels_first
    bob.learn.tensorflow.utils.image.blocks_tensorflow
    bob.learn.tensorflow.utils.image.tf_repeat
    bob.learn.tensorflow.utils.image.all_patches


Keras Utilities
===============

.. autosummary::
    bob.learn.tensorflow.utils.keras.SequentialLayer
    bob.learn.tensorflow.utils.keras.keras_channels_index
    bob.learn.tensorflow.utils.keras.keras_model_weights_as_initializers_for_variables
    bob.learn.tensorflow.utils.keras.restore_model_variables_from_checkpoint
    bob.learn.tensorflow.utils.keras.initialize_model_from_checkpoint
    bob.learn.tensorflow.utils.keras.model_summary
61 62


63 64 65 66 67 68 69 70 71 72 73
Math Utilities
==============

.. autosummary::
    bob.learn.tensorflow.utils.math.gram_matrix
    bob.learn.tensorflow.utils.math.upper_triangle_and_diagonal
    bob.learn.tensorflow.utils.math.upper_triangle
    bob.learn.tensorflow.utils.math.pdist
    bob.learn.tensorflow.utils.math.cdist
    bob.learn.tensorflow.utils.math.random_choice_no_replacement

74

Tiago de Freitas Pereira's avatar
Tiago de Freitas Pereira committed
75
Detailed Information
Amir MOHAMMADI's avatar
Amir MOHAMMADI committed
76 77 78
====================

.. automodule:: bob.learn.tensorflow
79 80
.. automodule:: bob.learn.tensorflow.data
.. automodule:: bob.learn.tensorflow.losses
81 82 83 84 85
.. automodule:: bob.learn.tensorflow.models
.. automodule:: bob.learn.tensorflow.utils
.. automodule:: bob.learn.tensorflow.utils.image
.. automodule:: bob.learn.tensorflow.utils.keras
.. automodule:: bob.learn.tensorflow.utils.math