当图像没有目标对象时,YOLO 推理崩溃

问题描述

假设YOLO模型是基于人训练的。我正在使用 Opencv DNN 进行推理。然后,我使用输出边界框进行进一步的图像处理以测量距离。

当我对没有目标对象的道路运行推理时,假设没有人,程序崩溃并给我一个错误。示例图像是:enter link description here

这是推理:

classes,scores,Boxes = model.detect(image,CONFIDENCE_THRESHOLD,NMS_THRESHOLD)


for (classid,score,Box) in zip(classes,Boxes):
    color = COLORS[int(classid) % len(COLORS)]
    cv2.rectangle(image,Box,color,2)


# other image processing steps
image,px,py=measurement.original(Box,image)

我的问题是:当现场没有人时,它会崩溃,因为它没有找到任何要检测的目标对象来进行测量,没有边界框。我该如何解决这个问题?并返回打印“未找到目标”作为输出。 我得到的错误

File "detection.py",line 72,in <module>
    image,image)
NameError: name 'Box' is not defined

PS:如果有边界框,它返回一个类似的值:[[1251 612 75 79]] 但是当没有边界框时它什么都不返回,空白

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)

相关问答

Selenium Web驱动程序和Java。元素在(x,y)点处不可单击。其...
Python-如何使用点“。” 访问字典成员?
Java 字符串是不可变的。到底是什么意思?
Java中的“ final”关键字如何工作?(我仍然可以修改对象。...
“loop:”在Java代码中。这是什么,为什么要编译?
java.lang.ClassNotFoundException:sun.jdbc.odbc.JdbcOdbc...