用ms rcnn訓練出的模型進行預測時提示缺乏工具

報錯信息以下:git

Traceback (most recent call last):
  File "demo/predict.py", line 8, in <module>
    from predictor import COCODemo
  File "/home/zhhu/wjy/mask_scoring_R-CNN/maskscoring_rcnn/demo/predictor.py", line 11, in <module>
    from maskrcnn_benchmark.utils import cv2_util
ImportError: cannot import name 'cv2_util'

解決辦法:github

去github下載maksrcnn-benchmark:https://github.com/facebookresearch/maskrcnn-benchmark。將這個util目錄下的cv2_util.py給拷貝到maskscoring_rcnn\maskrcnn_benchmark\utils下便可。spa

  

 

 

若是還報錯:code

Traceback (most recent call last):
  File "demo/predict.py", line 8, in <module>
    from predictor import COCODemo
  File "/home/zhhu/wjy/mask_scoring_R-CNN/maskscoring_rcnn/demo/predictor.py", line 411, in <module>
    from maskrcnn_benchmark.structures.keypoint import PersonKeypoints
ModuleNotFoundError: No module named 'maskrcnn_benchmark.structures.keypoint'

解決方法同理:blog

   

相關文章
相關標籤/搜索