Detectron2视频对象检测

Question

我使用detectron2运行obeject检测代码。我训练了网络，它在图像上工作得很好。现在我想在视频上测试它，但当我运行视频时，它没有检测到我的任何标签，更重要的是，它检测到了未经训练的对象，这意味着我的代码使用了其他一些模型路径。

# 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

我是这个领域的新手，所以我不知道我需要修改哪一行。我在哪里输入我训练过的模型？

谢谢

Answer 1

如果您阅读下面的教程，您将了解detectron2的基础知识。这是非常容易开始的。然后运行视频推理。

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