birdnet_tiny_forge.models package¶
Submodules¶
birdnet_tiny_forge.models.base module¶
Abstract Base Class for a ModelFactory, a simple object that can create a model. Meant to be registered with the model registry to make it easy to pick a model given runtime parameters.
- class birdnet_tiny_forge.models.base.ModelFactoryBase¶
Bases:
ABC- abstract classmethod create(class_count, input_size, **kwargs) keras.Model¶
- class birdnet_tiny_forge.models.base.TrainingSession(*args: Any, **kwargs: Any)¶
Bases:
Callback- CHKP_NAME = 'chkpt_{epoch:06d}.keras'¶
- SELECT_FNS = {'max': <function argmax>, 'min': <function argmin>}¶
- get_best_checkpoint(metric='val_loss', mode='min') keras.Model¶
Get the best checkpoint based on the specified metric
- get_metrics()¶
- on_epoch_end(epoch, logs=None)¶
- birdnet_tiny_forge.models.base.checkpoint_session(clean_up: bool = False)¶
birdnet_tiny_forge.models.mobilenet_slimmed module¶
Very simple MobileNet-inspired model
- class birdnet_tiny_forge.models.mobilenet_slimmed.MobilenetSlimmed¶
Bases:
ModelFactoryBase- classmethod create(class_count, input_shape, n_filters_1=32, n_filters_2=64, dropout=0.02)¶