JustCaptain kod-buduschego
27a7473e506a4bc52e95e3c8e2b1d9daa4087b34
Перейти к файлу
Код
main.py
from imageai.Detection import ObjectDetection
import os

detector = ObjectDetection()
model_path = "yolov3.pt"
detector.setModelTypeAsYOLOv3()
detector.setModelPath(model_path)
detector.loadModel()
print("START")

detections = detector.detectObjectsFromImage(
    input_image="image_fire.jpg",
    output_image_path="output_2.jpg",
    minimum_percentage_probability=30,
)