Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- 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"] )
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement