Пакетная обработка файлов

import os
from imageai.Detection import ObjectDetection
 
 
exec_path = os.getcwd()
 
detector = ObjectDetection()
detector.setModelTypeAsRetinaNet()
detector.setModelPath(os.path.join(
    exec_path, "resnet50_coco_best_v2.1.0.h5")
)
 
detections = detector.detectObjectsFromImage(
    input_image=os.path.join(exec_path, "input.jpg"),
    output_image_path=os.path.join(exec_path, "output.jpg"),
    minimum_percentage_probability=80,
    display_percentage_probability=True,
    display_object_name=False,
    display_box=True
)
 
 
for eachObject in detections:
    print(eachObject["name"], " : ", eachObject["percentage_probability"], " : ", eachObject["box_points"])
    print("--------------------------------")

import os
from imageai.Detection import ObjectDetection
 
exec_path = os.getcwd()
 
detector = ObjectDetection()
detector.setModelTypeAsRetinaNet()
detector.setModelPath(os.path.join(
    exec_path, "resnet50_coco_best_v2.1.0.h5")
)
detector.loadModel(detection_speed="flash")
custom_objects = detector.CustomObjects(car=True)
 
pth_in =  r'C:\Users\akakecha\PycharmProjects\imageAI\images_in'
pth_out =  r'C:\Users\akakecha\PycharmProjects\imageAI\images_out'
 
for file_name in os.listdir(pth_in):
    full_path = f'{pth_in}\{file_name}'
    full_path2 = f'{pth_out}\{file_name}'
 
detections = detector.detectObjectsFromImage(
    input_image=full_path,
    output_image_path=full_path2,
    minimum_percentage_probability=80,
    display_percentage_probability=True,
    display_object_name=False,
    display_box=True
)
 
 
for eachObject in detections:
    print(eachObject["name"], " : ", eachObject["percentage_probability"], " : ", eachObject["box_points"])
    print("--------------------------------")