Skip to content

Instantly share code, notes, and snippets.

@OlafenwaMoses

OlafenwaMoses/apple_detection_training.py

Created August 30, 2019 14:21
Show Gist options
  • Save OlafenwaMoses/2de7b8c7acaedf494216c9ec9d0ea655 to your computer and use it in GitHub Desktop.
Save OlafenwaMoses/2de7b8c7acaedf494216c9ec9d0ea655 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
apple_detection_training.py
from imageai.Detection.Custom import DetectionModelTrainer
trainer = DetectionModelTrainer()
trainer.setModelTypeAsYOLOv3()
trainer.setDataDirectory(data_directory="apple_dataset")
trainer.setTrainConfig(object_names_array=["apple", "damaged_apple"], batch_size=8, num_experiments=50, train_from_pretrained_model="pretrained-yolov3.h5")
trainer.trainModel()
@sy-15
Copy link

sy-15 commented May 3, 2020

How to generate loss curve while training the model?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment