Training models
Multiple CNN based backbones available for training
get_model
get_model (cfg)
Get an image classifier with a CNN based backbone.
Calls get_backbone
and adds a top_model layer to it.
Type | Details | |
---|---|---|
cfg | cfg (omegaconf.DictConfig) | Hydra Configuration |
Returns | tensorflow.keras.Model | Model object |
/opt/homebrew/Caskroom/miniconda/base/envs/rocks/lib/python3.9/site-packages/fastcore/docscrape.py:225: UserWarning: Unknown section Raises
else: warn(msg)
get_backbone
get_backbone (cfg)
Get backbone for the model.
List of supported models.
Type | Details | |
---|---|---|
cfg | cfg (omegaconf.DictConfig): | Hydra Configuration |
Returns | keras.api._v2.keras.models | Tensroflow Model |