Обнаружение видеообъектов Detectron2 - PullRequest
0 голосов
/ 04 августа 2020

Я запускаю код обнаружения объектов с помощью detectron2. Я тренировал net, и он отлично работает с изображениями. Теперь я хотел бы протестировать его на видео, но когда я запускаю видео, он не обнаруживает никаких моих ярлыков, более того, он обнаруживает необученный объект, что означает, что мой код использует какой-то другой путь к модели.

# Run frame-by-frame inference demo on this video (takes 3-4 minutes)
# Using a model trained
!git clone https://github.com/facebookresearch/detectron2
!cd detectron2 && python demo/demo.py --video-input ../video-clip.mp4 --confidence-threshold 0.6 -- 
output ../video-output-agripper.mkv
#!cd detectron2 && python demo/demo-test.py --config-file configs/Agripper/agripper.yaml --video- 
input ../video-clip.mp4 --confidence-threshold 0.6 --output ../video-output-agripper.mkv --opts 
MODEL.WEIGHTS configs/Agripper/model_0024999.pth

Я новичок в этом поле, поэтому не могу найти строку, которую мне нужно изменить. Где я ввожу свою обученную модель?

Спасибо

1 Ответ

0 голосов
/ 27 августа 2020

Если вы go изучите следующее руководство, вы узнаете основы работы с detectron2. Начать довольно просто. Затем запустите вывод видео.

https://colab.research.google.com/drive/16jcaJoc6bCFAQ96jDe2HwtXj7BMD_-m5

...