Укажите объект в YOLO

Question

Я использую YOLO для распознавания изображений.Я делаю снимок каждые полсекунды (потому что компьютер недостаточно быстр, чтобы использовать видео в качестве входных данных) и отправляю его в YOLO для обнаружения объекта.Он успешно обнаруживает объекты в нем и пишет теги на них.Тем не менее, я хочу, чтобы он поднимал некоторые флаги, например, распечатывал что-то, когда он обнаруживает кошку на изображении.

Есть ли способ указать определенный объект (например, кошку, собаку или телефон) в YOLO?

Answer 1

Если вам нужен только конкретный объект, например, шина, которую вы можете редактировать, но есть много проектов yolo, которыми я пользуюсь moovel , а файл веса содержит только 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"]
}