Detectron预训练模型下载中断导致无法运行

最新推荐文章于 2024-03-04 13:46:30 发布

一颗仙桃儿

最新推荐文章于 2024-03-04 13:46:30 发布

阅读量4.9k

点赞数

CC 4.0 BY-SA版权

文章标签：深度学习目标检测计算机视觉

本文链接：https://blog.youkuaiyun.com/qq_40310050/article/details/121321449

在尝试使用Detectron进行目标检测时，由于预训练模型R-50.pkl下载速度过慢导致中断。为解决此问题，可以手动下载模型并将其放置于正确位置。配置文件中需修改模型路径，指定到本地下载的预训练模型路径，这样就能正常运行了。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

因为要训练自己的数据集（我的是coco格式），再把一些配置修改完成后，运行

 python tools/train_net.py --cfg configs/12_2017_baselines/e2e_faster_rcnn_R-50-FPN_1x.yaml OUTPUT_DIR out_dir

要使用Faster Rcnn，所以在terminal中运行的时候，会显示在下载R-50.pkl
在这里插入图片描述
下不下来，速度太慢，报的错：意思就是下不下来

AssertionError: Target URL https://dl.fbaipublicfiles.com/detectron/ImageNetPretrained/MSRA/R-50.pkl appears to be downloaded to the local cache file /tmp/detectron-download-cache/ImageNetPretrained/MSRA/R-50.pkl, but the md5 hash of the local file does not match the reference (actual: a4041c4526ff960b0f18057f8d7502a4 vs. expected: b19105fdd7baafb40480251377de18f2). You may wish to delete the cached file and try again to trigger automatic download.

在这里插入图片描述
解决方案：