15 lines
538 B
Python
15 lines
538 B
Python
|
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, "ai_image.jpg"), output_image_path=os.path.join(execution_path , "ai_image.out.jpg"))
|
||
|
|
|
||
|
|
for eachObject in detections:
|
||
|
|
print(eachObject["name"] , " : " , eachObject["percentage_probability"] )
|