Last active
August 13, 2022 17:11
-
-
Save OlafenwaMoses/38e68a0f3bcb7350785b67d2e6bc06ce to your computer and use it in GitHub Desktop.
Object detection in 10 lines of code
This file contains 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 import ObjectDetection | |
import os | |
execution_path = os.getcwd() | |
detector = ObjectDetection() | |
detector.setModelTypeAsRetinaNet() | |
detector.setModelPath( os.path.join(execution_path , "resnet50_coco_best_v2.1.0.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"] ) |
Here's what the above code is doing:
- We first import the ObjectDetection class from the ImageAI library.
- We then create an instance of the ObjectDetection class and set the model type to RetinaNet.
- We then set the model path to the path of the RetinaNet model file that we downloaded earlier.
- We then load the model into the ObjectDetection class instance.
- We then call the detectObjectsFromImage method, passing in the input image path and the output image path.
- We then print the name of the object and the percentage probability that the object detected is the correct one.
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
To detect the object from the image from scratch using python; Click here I found the best article https://debuggingsolution.blogspot.com/2022/02/object-detection-from-scratch-in-python.html