【全步骤】Pytorch 搭建自己的Unet语义分割平台

0.说明

参考的是up主Bubbliiiing的教程

视频教程

博客参考

我自己的项目路径：D:\demo\unet-pytorch-main

初识语义分割，卷积神经网络可以进行特征提取，在获取到足够多的特征之后，进行反卷积，一层又一层地将图像重新放大

下面只讲述代码训练过程：

1.代码下载

代码资源下载链接：unet-pytorch

数据集下载：百度网盘

提取码: 44mk

2.数据集下载

数据集导入到项目文件中

3.权重下载

权重下载：bubbliiiing /unet-pytorch

4.运行设置

gpu带不动，我把cuda关了

5.运行train.py

带不动，根本带不动，准备用学校的服务器