如何抑制在detectObjectsFromImage()中生成输出图像?

时间:2019-09-27 12:21:16

标签: python-3.x keras

我正在使用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")

1 个答案:

答案 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"