一、环境

conda+pytorch1.2.0+cuda10.0+pycharm

二、训练内容

数据集：VOC2007

网络：U-net

功能：分割+图像分类

三、步骤

安装软件、框架、包、cuda（不安用cpu跑也可以），下载数据集、代码、权重文件等。。。

Pytorch-gpu环境配置_哔哩哔哩_bilibili

......

附件：资源

Github源码下载地址为：
https://github.com/bubbliiiing/unet-pytorch

四、问题解决

4.1 还需要下载的库

（1）tensorboard

开始从原始地址下，太慢太慢，找镜像源下载，试了阿里云没有合作，清华源有！！！

 （1）future

pip install future -i https://pypi.tuna.tsinghua.edu.cn/simple/

运行train.py时候如果报错：No module named 'past'

那么就是没有'past'包，但是根本没有这个名字，实际上是需要下载包future！！！

 4.1 开始训练，运行train.py

 终于开始了，但是数据集太庞大，10000多张图，太多了，查看GPU好像没有用到啊

补充一个查看这个的方法：win+i可以调出搜索：任务管理器——性能

 果然，不出所料，还是出错了？？？why why why

查询发现还是GPU用不上，于是乎改了CPU，跑起来了，哈哈哈哈.但是速度就惆怅了！

Epoch 1/100:   0%|          | 5/5291 [03:58<69:36:17, 47.40s/it, f_score=0.79, lr=1e-5, total_loss=0.226]

等不到可以先终止，接下来有时间再跑，或者想想办法搞上gpu！

 🆗🆗🆗🆗🆗🆗🆗🆗🆗🆗🆗🆗🆗🆗🆗🆗🆗🆗🆗🆗🆗🆗🆗🆗🆗🆗🆗🆗🆗🆗

五、测试

先选择用cpu跑一个Epoch的，loss虽然高点，但是可以出个结果。

1、修改unet.py,运行

 2、运行predict.py

 3、结果