基于Pytorch的U-Net调试记录

写在前面

一直想用基于pytorch的U-Net训练自己的数据集，这里的训练细节，我有很多的疑问。第一个疑问，训练使用的图片必须是方形的吗（例如512*512）？矩形的图片能不能训练？第二个疑问，如何标注训练的数据，比如需要标注多个类型，应该如何进行标注？

代码版本

我使用的代码是下面链接中代码。

https://github.com/milesial/Pytorch-UNet	

具体的版本是：Pytorch-Unet-3.0

如何将标注好的json数据转化为训练使用的图片

标注的时候，使用的是labelme，但是生成的都是json文件，如何将这些文件转化为训练使用的图片，也牵扯到了很多的细节。

训练需要的图片数据

训练数据只需要训练原图（对通道数量有要求吗？）和掩码图（对通道数量有要求吗？）就行。
将原图和掩码图分别放在data目录下的imgs文件夹和masks文件夹中。