我正在使用ImageAI的detectObjectsFromImage函数在某些图像上运行预先训练的keras模型。它可以正常工作,但是会生成带有针对对象的封闭框的输出图像。理想情况下,我不希望生成输出图像,因为我只需要检测代码并对其执行操作即可。我尝试不给出输出参数,但它仍然创建了一个jpg文件。
有什么想法可以抑制输出图像吗?
代码:
detections = detector.detectObjectsFromImage(input_image=os.path.join("pics",filename))
for eachObject in detections:
if(eachObject["name"]=="person"):
print("yes")
else:
print("no")
答案 0 :(得分:1)
将output_type
设置为"array"
:
detected_image_array, detections = detector.detectObjectsFromImage(output_type="array", input_image="image.jpg" ) # For numpy array output type
在文档中描述:https://github.com/OlafenwaMoses/ImageAI/blob/master/imageai/Detection/README.md
注意:您也可以将input_type
设置为"array"
。