使用自己数据集复现U-Net,R2U-Net,Att U-net,Att R2U-net过程。(代码参考: LeeJunHyun / Image_Segmentation )
主要就是dataset.py和dataloader.py里面改改就可以。main.py可以根据自己的需要修改相应的参数。注意自己使用的数据集的图片类型即可。文章目录前言一、LoadData二、Training1.DefineNeuralNetwork2.Lossfunction3.OptimizationAlgorithm三、Validation四、Testing五、EvaluationMain.