Skip to content

Instantly share code, notes, and snippets.

@OlafenwaMoses

OlafenwaMoses/custom_detection_model_evaluation.py

Created August 1, 2019 13:21
Show Gist options
  • Save OlafenwaMoses/99dfd9522145926e828e956bb6678e1b to your computer and use it in GitHub Desktop.
Save OlafenwaMoses/99dfd9522145926e828e956bb6678e1b to your computer and use it in GitHub Desktop.
Download ZIP
Code to evaluate saved detection models using ImageAI
Raw
custom_detection_model_evaluation.py
from imageai.Detection.Custom import DetectionModelTrainer
trainer = DetectionModelTrainer()
trainer.setModelTypeAsYOLOv3()
trainer.setDataDirectory(data_directory="hololens")
trainer.evaluateModel(model_path="hololens/models", json_path="hololens/json/detection_config.json", iou_threshold=0.5, object_threshold=0.3, nms_threshold=0.5)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment