# pytorch-unet

**Repository Path**: qiaofengsheng/pytorch-unet

## Basic Information

- **Project Name**: pytorch-unet
- **Description**: pytorch搭建自己的unet网络，训练自己的数据集。
B站视频地址全程带你手撸代码：https://www.bilibili.com/video/BV11341127iK?spm_id_from=333.999.0.0
- **Primary Language**: Python
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 57
- **Forks**: 22
- **Created**: 2022-01-13
- **Last Updated**: 2025-06-17

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# pytorch-unet

#### 介绍
pytorch搭建自己的unet网络，训练自己的数据集。
B站视频地址全程带你手撸代码：https://www.bilibili.com/video/BV11341127iK?spm_id_from=333.999.0.0

#### 软件架构
pythorch


#### 安装教程

1.  下载最新版本的pytorch就可以

#### 使用说明

1.  数据集原图存放地址：data/JPEGImages   mask存放地址：data/SegmentationClass
2.  直接运行train.py,其中train_image文件夹存储的是训练过程中的效果图
3.  params文件夹保存权重
4.  测试test.py，用来测试图片，测试结果存储在result文件夹中


#### 视频地址

B站：https://www.bilibili.com/video/BV11341127iK?spm_id_from=333.999.0.0