更换路径
pytorch中封装了很多实现,需要用到官方下载的数据集,如果本地有下载好的就不需要再下载了
https://www.freesion.com/article/5198895969/
看这里,改路径然后就可以用本地的了
运行时出现
GPU的坑
GPU的环境没装好,所以如果选择GPU环境,模型跑不起来
测试语句:
两个都输出说明当前环境的cuda可以正常使用
否则:
只能用CPU环境,
带镜像安装库
pip install opencv==3.4.2 -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com
分而治之思想
模型训练时读取数据集格式,
可以先了解正确的读取格式
再写个脚本直接读,读正确了再放到模型中进行读取