在YOL中指定对象

Question

我用YOLO做图像识别。我每半秒钟拍一张照片（因为电脑速度不够快，无法将视频作为输入），然后把它发送给YOLO来检测物体。它成功地检测到其中的对象并在其上写入标记。但是，当它在图像中检测到猫时，我希望它能升起一些标志，比如打印出一些东西。你知道吗

有没有办法在YOLO中指定某个对象（比如猫、狗或电话）？你知道吗

Answer 1

如果您只需要一个特定的对象，如总线，您可以编辑 但是我使用了很多yolo项目moovel，权重文件只包含24个类，权重文件名是yolo-挥发性有机物重量所以你可以试试 [aeroplane bicycle bird boat bottle bus car cat chair cow diningtable dog horse motorbike person pottedplant sheep sofa train tvmonitor ]

这就是配置.json文件可以在VALID_CLASSES中添加此总线或汽车，如果需要该类中的所有内容，请使用"VALID_CLASSES": ["*"]

{
  "PATH_TO_YOLO_DARKNET" : "/home/nvidia/darknet-net",
  "VALID_CLASSES": ["bus"]
}