我想知道如何使用类似YOLO的技术来检测缺失的对象?假设我有一个视频,拍摄了智能手机、钥匙和护照。然后我有另一个视频,只拍摄了智能手机和钥匙,是否可以发出警告,指出有一个对象缺失?
回答:
可以。如果你使用的是AlexeyAB的darknet的Python版本,你可以在这一行看到所有的检测结果:
https://github.com/AlexeyAB/darknet/blob/master/darknet_video.py#L81
然后在数组中搜索标签,如果某个标签不在该数组中,你可以返回一个警告或执行你需要的任何操作。
if b'smartphone' not in detections[0]:
print('Warning')