情况:运行Mask R-CNN_demo 报错,文件路径: /maskrcnn/maskrcnn-benchmark-master/demo/Mask_R-CNN_demo.ipynb
检测没有权重文件后会自动下载权重文件,但如果中间断开后则会报错UnpicklingError: pickle data was truncated,且权重文件无法继续下载。
预训练的权重会存在~/.torch里,比如~/.torch/model (前提是你已经训练过一次了,第一次训练会自动从网上下载)
解决办法:cd ~/.torch/model
删除目录中已有的权重文件,重新运行该demo即可。
coco_demo = COCODemo(
cfg,
min_image_size=800,
confidence_threshold=0.7,
)