Question

我使用detectron2运行对象检测代码。我训练了网络，并且可以很好地处理图像。现在，我想在视频上对其进行测试，但是当我运行视频时，它不会检测到我的任何标签，更重要的是，它检测到未训练的对象，这意味着我的代码使用其他模型路径。

# 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

Detectron2视频对象检测

1 个答案: