pytorch-unet
介绍
pytorch搭建自己的unet网络,训练自己的数据集。
软件架构
pythorch
安装教程
- 下载最新版本的pytorch就可以
使用说明
- 数据集原图存放地址:data/JPEGImages mask存放地址:data/SegmentationClass
- 直接运行train.py,其中train_image文件夹存储的是训练过程中的效果图
- params文件夹保存权重
- 测试test.py,用来测试图片,测试结果存储在result文件夹中
视频地址
B站:https://www.bilibili.com/video/BV11341127iK?spm_id_from=333.999.0.0