Author: Danelle Cline, [email protected]

About

• kclassify * Tensorflow-Keras image classifier for transfer-learning training either locally or in AWS SageMaker

This trains an image classifier using transfer learning with the Tensorflow Keras library with choices of

Optimizers

• Radam
• Adam
• Ranger (not working as of 7-20-21)

Models

• efficientnetB0 EfficientNetB0
• resnet50
• vgg16
• vgg19
• mobilenetv2

Augmentations

• width, shift, and zoom
• horizontal/vertical flip

and all the typical hyperparameters needed for model training like learning rate, batch size, etc.

Documentation can be found here.

Requirements

• Docker
• AWS Account

Questions?

If you have any questions or are interested in contributing, please contact me at [email protected].

Danelle Cline https://www.mbari.org/cline-danelle-e/

---