py_api.rst 3.26 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

Amir MOHAMMADI's avatar
Amir MOHAMMADI committed
10 11
Estimators
==========
Tiago de Freitas Pereira's avatar
Tiago de Freitas Pereira committed
12

Amir MOHAMMADI's avatar
Amir MOHAMMADI committed
13 14
.. autosummary::
    bob.learn.tensorflow.estimators.Logits
15 16 17
    bob.learn.tensorflow.estimators.LogitsCenterLoss
    bob.learn.tensorflow.estimators.Triplet
    bob.learn.tensorflow.estimators.Siamese
18
    bob.learn.tensorflow.estimators.Regressor
19 20
    bob.learn.tensorflow.estimators.MovingAverageOptimizer
    bob.learn.tensorflow.estimators.learning_rate_decay_fn
21 22


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

Amir MOHAMMADI's avatar
Amir MOHAMMADI committed
24
Architectures
25 26 27 28 29 30 31
=============

.. autosummary::
    bob.learn.tensorflow.network.chopra
    bob.learn.tensorflow.network.light_cnn9
    bob.learn.tensorflow.network.dummy
    bob.learn.tensorflow.network.mlp
32
    bob.learn.tensorflow.network.mlp_with_batchnorm_and_dropout
33 34 35 36
    bob.learn.tensorflow.network.inception_resnet_v2
    bob.learn.tensorflow.network.inception_resnet_v1
    bob.learn.tensorflow.network.inception_resnet_v2_batch_norm
    bob.learn.tensorflow.network.inception_resnet_v1_batch_norm
Amir MOHAMMADI's avatar
Amir MOHAMMADI committed
37
    bob.learn.tensorflow.network.SimpleCNN.slim_architecture
38 39
    bob.learn.tensorflow.network.vgg_19
    bob.learn.tensorflow.network.vgg_16
Tiago de Freitas Pereira's avatar
Tiago de Freitas Pereira committed
40

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

Amir MOHAMMADI's avatar
Amir MOHAMMADI committed
42 43
Data
====
Tiago de Freitas Pereira's avatar
Tiago de Freitas Pereira committed
44

Amir MOHAMMADI's avatar
Amir MOHAMMADI committed
45 46
.. autosummary::
    bob.learn.tensorflow.dataset.bio.BioGenerator
47 48 49 50 51
    bob.learn.tensorflow.dataset.image.shuffle_data_and_labels_image_augmentation
    bob.learn.tensorflow.dataset.siamese_image.shuffle_data_and_labels_image_augmentation
    bob.learn.tensorflow.dataset.triplet_image.shuffle_data_and_labels_image_augmentation
    bob.learn.tensorflow.dataset.tfrecords.shuffle_data_and_labels_image_augmentation
    bob.learn.tensorflow.dataset.tfrecords.shuffle_data_and_labels
52
    bob.learn.tensorflow.dataset.generator.dataset_using_generator
Amir MOHAMMADI's avatar
Amir MOHAMMADI committed
53 54
    bob.learn.tensorflow.utils.util.to_channels_last
    bob.learn.tensorflow.utils.util.to_channels_first
Tiago de Freitas Pereira's avatar
Tiago de Freitas Pereira committed
55

56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76

Style Transfer
==============

.. autosummary::
    bob.learn.tensorflow.style_transfer.do_style_transfer


Losses
======

.. autosummary::
    bob.learn.tensorflow.loss.mean_cross_entropy_loss
    bob.learn.tensorflow.loss.mean_cross_entropy_center_loss
    bob.learn.tensorflow.loss.contrastive_loss
    bob.learn.tensorflow.loss.triplet_loss
    bob.learn.tensorflow.loss.triplet_average_loss
    bob.learn.tensorflow.loss.triplet_fisher_loss
    bob.learn.tensorflow.loss.linear_gram_style_loss
    bob.learn.tensorflow.loss.content_loss
    bob.learn.tensorflow.loss.denoising_loss
77 78
    bob.learn.tensorflow.loss.balanced_softmax_cross_entropy_loss_weights
    bob.learn.tensorflow.loss.balanced_sigmoid_cross_entropy_loss_weights
79 80 81



82

Tiago de Freitas Pereira's avatar
Tiago de Freitas Pereira committed
83
Detailed Information
Amir MOHAMMADI's avatar
Amir MOHAMMADI committed
84 85 86 87 88
====================

.. automodule:: bob.learn.tensorflow
.. automodule:: bob.learn.tensorflow.estimators
.. automodule:: bob.learn.tensorflow.dataset
Amir MOHAMMADI's avatar
Amir MOHAMMADI committed
89
.. automodule:: bob.learn.tensorflow.dataset.generator
Amir MOHAMMADI's avatar
Amir MOHAMMADI committed
90
.. automodule:: bob.learn.tensorflow.dataset.bio
91 92 93 94 95
.. automodule:: bob.learn.tensorflow.dataset.image
.. automodule:: bob.learn.tensorflow.dataset.siamese_image
.. automodule:: bob.learn.tensorflow.dataset.triplet_image
.. automodule:: bob.learn.tensorflow.dataset.tfrecords
.. automodule:: bob.learn.tensorflow.network
96
.. automodule:: bob.learn.tensorflow.network.SimpleCNN
Amir MOHAMMADI's avatar
Amir MOHAMMADI committed
97 98
.. automodule:: bob.learn.tensorflow.utils
.. automodule:: bob.learn.tensorflow.utils.util
99 100
.. automodule:: bob.learn.tensorflow.style_transfer
.. automodule:: bob.learn.tensorflow.loss