基于pytorch的U-net模型简单复现

自己是计算机方向研一的学生,打算做的方向是图像分割方面,又因为u-net也是做医学图像方面很基础的网络,于是写个文章记录下自己复现别人代码的过程,学习借鉴一下。同时有错误也请各位大佬批评指正。

关于U-net具体的代码以及相关过程参考了以下这篇博客:
Pytorch入门——用UNet网络做图像分割



整体的文件目录

在这里插入图片描述


一、一些参数的调整

在这里插入图片描述
1、先把action中的默认值设置为train,随后跑完得到模型权重

评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值