(已解决)运行Mask R-CNN_demo 报错 UnpicklingError: pickle data was truncated 求指导～

情况：运行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,
)