自己是计算机方向研一的学生,打算做的方向是图像分割方面,又因为u-net也是做医学图像方面很基础的网络,于是写个文章记录下自己复现别人代码的过程,学习借鉴一下。同时有错误也请各位大佬批评指正。 关于U-net具体的代码以及相关过程参考了以下这篇博客: Pytorch入门——用UNet网络做图像分割 文章目录 整体的文件目录 一、一些参数的调整 二、训练过程 1.训练的结果 2.遇到的问题 3.验证结果 4.总结 整体的文件目录 一、一些参数的调整 1、先把action中的默认值设置为train,随后跑完得到模型权重