b2eaa8da9b02e1bfb94cf3e9a886e8d0158a4f36
Перейти к файлуt
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,
)