JustCaptain kod-buduschego
llm/middle-9c3d46
Перейти к файлу
Код
main.py
from imageai.Detection import ObjectDetection

print("STEP 1 | DETECTOR OBJECT")
detector = ObjectDetection()
model_path = "yolov3.pt"
detector.setModelTypeAsYOLOv3()
detector.setModelPath(model_path)
detector.loadModel()

print("STEP 2 | SEARCH")
detections = detector.detectObjectsFromImage(
    input_image="input_image.jpg",
    output_image_path="output_image.jpg",
    minimum_percentage_probability=30,
)