1

我正在尝试让Detectron2的演示代码在我的笔记本电脑上本地工作。一切似乎都运行正常，但没有检测到对象实例，即使我使用 Colab 演示中的图像也是如此。

我在非 GPU Mac 上运行。我按照安装说明安装了 Detectron。我的机器上有以下模块版本：

  • detectron2@git+https://github.com/facebookresearch/detectron2.git@ea3b3f22bf1de58008599794f149149ff65d3780
  • opencv-python==4.5.3.56
  • 火炬==1.9.0
  • 火炬视觉==0.10.0

我从 Detectron 的 github 复制了 demo.py、predictor.py、mask_rcnn_R_101_FPN_3x.yaml 和 Base-RCNN-FPN.yaml。然后我使用预训练模型命令运行推理演示。具体命令是这样的：

python demo.py --input 000000439715.jpeg --output output --config-file mask_rcnn_R_101_FPN_3x.yaml --opts MODEL.WEIGHTS detectron2://COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x/137849600/model_final_f10217.pkl MODEL.DEVICE cpu

000000439715.jpeg 是来自Colab 笔记本演示的骑马人的示例图像。输出的最后一行是

000000439715.jpeg: detected 0 instances in 6.77s

输出目录中的图像上没有注释。

日志输出对我来说看起来不错。唯一可能表明存在问题的是顶部的警告

[08/28 12:35:18 detectron2]: Arguments: Namespace(confidence_threshold=0.5, config_file='mask_rcnn_R_101_FPN_3x.yaml', input=['000000439715.jpeg'], opts=['MODEL.WEIGHTS', 'detectron2://COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x/137849600/model_final_f10217.pkl', 'MODEL.DEVICE', 'cpu'], output='output', video_input=None, webcam=False)
[08/28 12:35:18 fvcore.common.checkpoint]: [Checkpointer] Loading from detectron2://COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x/137849600/model_final_f10217.pkl ...
[08/28 12:35:18 fvcore.common.checkpoint]: Reading a file from 'Detectron2 Model Zoo'
WARNING [08/28 12:35:19 fvcore.common.checkpoint]: Some model parameters or buffers are not found in the checkpoint:

我不知道该怎么做。

我尝试不指定模型权重。我还尝试将置信度阈值设置为零。我得到了同样的结果。

难道我做错了什么？接下来的调试步骤是什么？

4

0 回答 0