在运行程序的时候碰到的问题。因为网络环境不好,导致下载的faster-rcnn的预训练模型中途下载失败。在网上找了很久解决方案,说是删除下载失败的文件重新下载,但是我找不到那个失败文件,也没有找到checkpoints文件。
于是在/usr/local/anaconda3/lib/python3.8/site-packages/torch这个路径下找到了hub.py文件,在load_state_dict_from_url方法中,model_dir是设置预训练文件存储路径的,默认为None,重新设置一个文件夹存储即可。
在方法中添加
model_dir='/usr/local/anaconda3/lib/python3.8/site-packages/torch/checkpoints'