# 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
U-net模型调整优化（引入残差网络及小波变换模块）
需积分: 0 179 浏览量
2024-05-22
18:47:28
上传
评论
收藏 135.92MB RAR 举报
U-net模型调整优化（引入残差网络及小波变换模块） （128个子文件） 
.gitignore 2KB .gitignore 47B pytorch-unet.iml 321B LICENSE 51B README.en.md 984B README.md 730B README.md 18B unet.pth 122.5MB net.py 4KB train.py 1KB data.py 912B test.py 547B utils.py 251B net.cpython-36.pyc 3KB data.cpython-36.pyc 1KB utils.cpython-36.pyc 481B 10.tif 768KB 46.tif 591KB 7.tif 591KB 73.tif 591KB 91.tif 591KB 39.tif 591KB 64.tif 591KB 21.tif 591KB 65.tif 591KB 81.tif 591KB 56.tif 591KB 71.tif 591KB 84.tif 591KB 15.tif 591KB 88.tif 591KB 70.tif 591KB 53.tif 591KB 54.tif 591KB 14.tif 591KB 37.tif 591KB 79.tif 591KB 86.tif 591KB 23.tif 591KB 18.tif 591KB 92.tif 591KB 12.tif 591KB 95.tif 591KB 74.tif 591KB 9.tif 591KB 1.tif 591KB 59.tif 591KB 13.tif 591KB 57.tif 591KB 72.tif 591KB 77.tif 591KB 51.tif 591KB 75.tif 591KB 24.tif 591KB 25.tif 591KB 16.tif 591KB 99.tif 591KB 94.tif 591KB 50.tif 591KB 31.tif 591KB 58.tif 591KB 66.tif 591KB 32.tif 591KB 89.tif 591KB 93.tif 591KB 38.tif 591KB 34.tif 591KB 80.tif 591KB 20.tif 591KB 26.tif 591KB 87.tif 591KB 82.tif 591KB 33.tif 591KB 47.tif 591KB 10.tif 591KB 0.tif 591KB 4.tif 591KB 69.tif 591KB 83.tif 591KB 63.tif 591KB 36.tif 591KB 27.tif 591KB 5.tif 591KB 8.tif 591KB 61.tif 591KB 98.tif 591KB 52.tif 591KB 22.tif 591KB 45.tif 591KB 44.tif 591KB 2.tif 591KB 100.tif 591KB 101.tif 591KB 78.tif 591KB 30.tif 591KB 62.tif 591KB 28.tif 591KB 76.tif 591KB 97.tif 591KB 67.tif 591KB共 128 条
- 1
- 2
资源评论
