Created
August 1, 2019 12:55
-
-
Save OlafenwaMoses/6c948aa66ff19aeaa9fa6d4f5f462981 to your computer and use it in GitHub Desktop.
Code for training custom object detection model with ImageAI
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| from imageai.Detection.Custom import DetectionModelTrainer | |
| trainer = DetectionModelTrainer() | |
| trainer.setModelTypeAsYOLOv3() | |
| trainer.setDataDirectory(data_directory="hololens") | |
| trainer.setTrainConfig(object_names_array=["hololens"], batch_size=4, num_experiments=100, train_from_pretrained_model="pretrained-yolov3.h5") | |
| trainer.trainModel() |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
It giving an error
i can give object name correctly but it shows file not found error pls clarify it
FileNotFoundError: [Errno 2] No such file or directory: 'Number plate/train/annotations'
