keras_retinanet.callbacks package

Submodules

keras_retinanet.callbacks.coco module

keras_retinanet.callbacks.common module

class keras_retinanet.callbacks.common.RedirectModel(callback, model)[source]

Bases: sphinx.ext.autodoc.importer._MockObject

Callback which wraps another callback, but executed on a different model.

`python model = keras.models.load_model('model.h5') model_checkpoint = ModelCheckpoint(filepath='snapshot.h5') parallel_model = multi_gpu_model(model, gpus=2) parallel_model.fit(X_train, Y_train, callbacks=[RedirectModel(model_checkpoint, model)]) `

Args
callback : callback to wrap. model : model to use when executing callbacks.
on_batch_begin(batch, logs=None)[source]
on_batch_end(batch, logs=None)[source]
on_epoch_begin(epoch, logs=None)[source]
on_epoch_end(epoch, logs=None)[source]
on_train_begin(logs=None)[source]
on_train_end(logs=None)[source]

keras_retinanet.callbacks.eval module

Module contents