Skip to content

Instantly share code, notes, and snippets.

@Shohjahon

Shohjahon/FirstDetection.py

Created December 9, 2018 15:58
Show Gist options
  • Save Shohjahon/7ef52681914e1abaf781781ba12a3c75 to your computer and use it in GitHub Desktop.
Save Shohjahon/7ef52681914e1abaf781781ba12a3c75 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
FirstDetection.py
from imageai.Detection import ObjectDetection
import os
execution_path = os.getcwd()
detector = ObjectDetection()
detector.setModelTypeAsRetinaNet()
detector.setModelPath( os.path.join(execution_path , "resnet50_coco_best_v2.0.1.h5"))
detector.loadModel()
detections = detector.detectObjectsFromImage(input_image=os.path.join(execution_path , "image.jpg"), output_image_path=os.path.join(execution_path , "imagenew.jpg"))
for eachObject in detections:
print(eachObject["name"] , " : " , eachObject["percentage_probability"] )
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment