Skip to content

Instantly share code, notes, and snippets.

@Praveen76

Praveen76/Train RetinaNet model on google colab.py

Created July 29, 2020 05:34
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save Praveen76/90aafd3f5b73a967a5eb9105d9df854d to your computer and use it in GitHub Desktop.
Save Praveen76/90aafd3f5b73a967a5eb9105d9df854d to your computer and use it in GitHub Desktop.
Download ZIP
Raw
Train RetinaNet model on google colab.py
#Put your training data path & file that has labels for your training data
!keras_retinanet/bin/train.py
--freeze-backbone
--random-transform
--weights {PRETRAINED_MODEL}
--batch-size 8
--steps 500
--epochs 15
csv maskDetectorData.csv
maskDetectorClasses.csv
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment