在YOLO中指定对象

Question

我正在使用YOLO进行图像识别。我每半秒钟拍摄一张照片（因为计算机的速度不足以将视频用作输入），并将其发送给YOLO以检测物体。它可以成功检测其中的对象并在其上写入标签。但是，我希望它在检测到图像中的猫时升起一些标志，例如打印出一些东西。

是否可以在YOLO中指定某个对象（例如猫，狗或电话）？

Answer 1

如果仅需要特定对象（例如总线），则可以进行编辑 但是我正在使用很多moovel的yolo项目，并且权重文件仅包含24个类，权重文件名为yolo-voc.weights，因此您可以尝试一下 [aeroplane bicycle bird boat bottle bus car cat chair cow diningtable dog horse motorbike person pottedplant sheep sofa train tvmonitor ]

这是config.json文件，如果您需要该类中的所有内容，则可以在VALID_CLASSES中添加此公交车或汽车，然后使用"VALID_CLASSES": ["*"]

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