问题描述
我想创建一个简单的对象检测项目。我想解析一个包含多张图像的 Numpy 数组,而不是将一个图像解析为 .detectObjectsFromImage()
。我创建了一个在图像数组内循环的 for 循环,我将输入选择为 images[x]
。我似乎因为 output_image_path
而收到错误消息。如果我想在同一个文件夹的不同文件中输出不同的图像,输出路径应该如何?
from imageai.Detection import ObjectDetection
import glob
import cv2
detector = ObjectDetection()
model_path = "/Users/dumitrescucristian/Desktop/Object_detection/models/yolo-tiny.h5"
output_path = "/Users/dumitrescucristian/Desktop/Object_detection/output"
images = [cv2.imread(file) for file in glob.glob("/Users/dumitrescucristian/Desktop/Object_detection/input/*.jpg")]
detector.setModelTypeAsTinyYOlov3()
detector.setModelPath(model_path)
detector.loadModel()
for x in range(len(images)):
detection[x] = detector.detectObjectsFromImage(input_image=images[x],output_image_path=output_path)
解决方法
我通过降级到 tensorflow 2.4.1 和 python