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

9月25日晚8点，Gitee 吐槽大会正式开幕，吐槽火力全开，Gitee 接招不慌！

forked from qiaofengsheng/pytorch-unet

加入 Gitee

与超过 1200万开发者一起发现、参与优秀开源项目，私有仓库也完全免费：）

免费加入

克隆/下载

分支 1

标签 0

2022年01月13日添加params和train_image文件夹 303e89b

4 次提交

.idea

data

params

result

test_image

train_image

.gitignore

LICENSE

README.en.md

README.md

data.py

net.py

test.py

train.py

utils.py

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

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

Python

取消

暂无发行版

马建仓 AI 助手

尝试更多

代码解读

代码找茬

代码优化

Python

https://gitee.com/jieagle/pytorch-unet.git

git@gitee.com:jieagle/pytorch-unet.git

jieagle

pytorch-unet

master

刘俊杰/pytorch-unet

简介

发行版

贡献者

近期动态

刘俊杰/pytorch-unet .gitee-modal { width: 500px !important; }

简介

发行版

贡献者

近期动态

搜索帮助

刘俊杰/pytorch-unet