目录
之前简单地写了一个pytorch的U-net 复现过程,有很多小伙伴在评论里有很多疑问,抽空又复现了一遍,简单整理了常见的问题。
之前写的教程:U-net复现pytorch版本 以及制作自己的数据集并训练_candice5566的博客-CSDN博客
---------------------------------------------------------------------------------------------------------------------
2021.11.14复现过程:
github代码链接:
https://github.com/milesial/Pytorch-UNet
代码说明:这个代码新加入了一个wandb的可视化库,能够可视化训练过程中的参数变化情况
1.下载github的代码,并且配置好环境:
2.下载数据集:
我自