ipynb.txt

pip install tensorflow==2.4.0

pip install opencv-python

pip install keras

pip install imageAI

from imageai.Detection import ObjectDetection

detector = ObjectDetection()

import urllib.request
urllib.request.urlretrieve(
    'https://gitlab.com/andreass.bayu/imageai-repo/-/raw/master/yolo-tiny.h5', 'yolo-tiny.h5')


model_path = "yolo-tiny.h5"
input_path = "./image/image2.jpg"
output_path = "image-yolo-tiny.jpg"

detector.setModelTypeAsTinyYOLOv3()

detector.setModelPath(model_path)

detector.loadModel()

detection = detector.detectObjectsFromImage(input_image=input_path, output_image_path=output_path)

for eachItem in detection:
    print(eachItem["name"] , " : ", eachItem["percentage_probability"])

from IPython.display import Image
Image(filename='image-yolo-tiny.jpg')