运行GitHub上的预训练模型--FaceBoxes.pytorch项目

运行Faceboxes.pytorch项目中的test.py遇到的问题

Faceboxes.pytorch项目链接

问题一:缺少cv2模块

ModuleNotFoundError: No module named 'cv2'

解决办法:
Anaconda缺少什么模块就安装什么模块,打开Anaconda prompt
输入:conda install --channel https://conda.anaconda.org/menpo opencv

参考链接

第一次更新失败,因为我的conda未更新,所以我要先更新conda再继续安装cv2模块

问题二:conda需要更新

RemoveError: ‘requests’ is a dependency of conda and cannot be removed from conda’s operating enviro…

解决办法:
输入 conda update --force conda

参考链接

等待conda 顺利更新结束,重新按照问题一的解决办法安装cv2

问题三:缺少utils

ModuleNotFoundError: No module named 'utils'

解决办法:Anaconda缺少什么模块就安装什么模块,打开Anaconda prompt
输入:conda install -c conda-forge python-utils

参考链接

问题四:该模型需要在GPU上运行,但是我的电脑只有CPU,且为AMD显卡无法安装CUDA.

ValueError: Expected a cuda device, but got: cpu

有考虑过是否要换一台电脑,但无奈自己平时用的电脑在学校没带回来,也考虑过要连接远程服务器来解决没有GPU的问题,但是自己没有使用过远程服务器,所以这个办法也是远水救近火,后面经“高人”指点:这些模型肯定是既可以在CPU上运行又可以在GPU上运行的,只不过你要将模型中选择用GPU运行的部分全部改成用CPU运行就行。所以我开始大概的浏览了一遍模型,将网络模型中用cuda的部分都改成了cpu,从而解决了这个问题。

我的解决办法:

  1. 在模型的前面添加:
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
  1. 将原模型中的这一句中的“cpu”的默认改成True
 parser.add_argument('--cpu', action="store_true", default=False, help=
  • 0
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 5
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值