- 检查配置文件中的num_classes(也就是数据集的num_classes)和代码中的是否一致,例如vaihingen和potsdam数据集的类别数量不一致
例如:num_classes用变量传入,不要写死
nn.Conv2d(512, self.num_classes, kernel_size=1, stride=1, padding=0, bias=True)
- 检查图片大小是否符合,例如有的数据增强方法是对potsdam数据集中的训练集裁剪为768,如果代码要求512尺寸,则会报错