U-Net调试记录
写在前面
一直想用基于pytorch的U-Net训练自己的数据集,这里的训练细节,我有很多的疑问。第一个疑问,训练使用的图片必须是方形的吗(例如512*512)?矩形的图片能不能训练?第二个疑问,如何标注训练的数据,比如需要标注多个类型,应该如何进行标注?
代码版本
我使用的代码是下面链接中代码。
https://github.com/milesial/Pytorch-UNet
具体的版本是:Pytorch-Unet-3.0
如何将标注好的json数据转化为训练使用的图片
标注的时候,使用的是labelme,但是生成的都是json文件,如何将这些文件转化为训练使用的图片,也牵扯到了很多的细节。
训练需要的图片数据
训练数据只需要训练原图(对通道数量有要求吗?)和掩码图(对通道数量有要求吗?)就行。
将原图和掩码图分别放在data目录下的imgs文件夹和masks文件夹中。