0

我使用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

我是这个领域的新手,所以我找不到需要更改的行。我在哪里输入训练有素的模型?

谢谢

4

1 回答 1

0

如果您通过以下教程,您将了解有关Detectron2的基础知识。这很容易开始。然后运行视频推理。

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

于 2020-08-27T01:36:24.937 回答