一、环境

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！

pytorch训练第一个项目VOC2007分割相关推荐

【项目实战课】基于Pytorch的Semantic_Human_Matting（人像软分割）实战
欢迎大家来到我们的项目实战课,本期内容是<基于Pytorch的Semantic_Human_Matting(人像软分割)实战>.所谓项目实战课,就是以简单的原理回顾+详细的项目实战的模式, ...
深度学习工程实践 6. 使用pytorch训练自己的眼球分割模型
深度学习工程实践 6. 使用pytorch训练自己的眼球分割模型 1. 概述 2. 目标 3. 工程实践 3.1 数据寻找,数据标注 3.2 训练 3.3 部署应用到桌面程序 4. 总结 1. 概述 ...
PyTorch训练加速17种技巧
点击上方"小白学视觉",选择加"星标"或"置顶" 重磅干货,第一时间送达文自机器之心作者:LORENZ KUHN 编辑:陈萍掌握这 ...
憨批的语义分割重制版9——Pytorch 搭建自己的DeeplabV3+语义分割平台
憨批的语义分割重制版9--Pytorch 搭建自己的DeeplabV3+语义分割平台注意事项学习前言什么是DeeplabV3+模型代码下载 DeeplabV3+实现思路一.预测部分 1.主干 ...
憨批的语义分割重制版6——Pytorch 搭建自己的Unet语义分割平台
憨批的语义分割重制版6--Pytorch 搭建自己的Unet语义分割平台注意事项学习前言什么是Unet模型代码下载 Unet实现思路一.预测部分 1.主干网络介绍 2.加强特征提取结构 3. ...
Jetson Nano目标检测手把手实战教程（pytorch训练、tensorrt推理，含完整代码和数据）
目录一.概述 1.1 深度学习和边缘计算 1.2 Jetson Nano简介二.深度学习环境安装 2.1 Pytorch框架 2.2 在Jetson Nano上安装Pytorch 三.算法原理四 ...
送你9个快速使用Pytorch训练解决神经网络的技巧（附代码）
来源:读芯术本文约4800字,建议阅读10分钟. 本文为大家介绍9个使用Pytorch训练解决神经网络的技巧图片来源:unsplash.com/@dulgier 事实上,你的模型可能还停留在石器时 ...
这17 种方法让 PyTorch 训练速度更快！
点上方蓝字计算机视觉联盟获取更多干货在右上方 ··· 设为星标 ★,与你不见不散仅作学术分享,不代表本公众号立场,侵权联系删除转载于:选自 | efficientdl.com 作者 | LO ...
pytorch训练GAN的代码（基于MNIST数据集）
论文:Generative Adversarial Networks 作者:Ian J. Goodfellow 年份:2014年从2020年3月多开始看网络,这是我第一篇看并且可以跑通代码的论文,简 ...

pytorch训练第一个项目VOC2007分割