语义分割标签必须是单通道、整型、像素值为离散类别ID且尺寸与原图严格对齐的图像；需避免RGB输入、浮点型数据、连续灰度值及尺寸错位，加载时须同步变换并验证唯一值、形状与类型。
语义分割模型训练时，标签不是随便画个图就行——它必须是**单通道、整型、像素值对应类别ID**的图像，且尺寸要和原图严格对齐。设计错格式，模型要么报错，要么学偏，后期排查极耗时。
标签图像的核心格式要求
标签本质是一张“类别地图”，每个像素的数值代表该位置属于哪一类（如0=背景，1=人，2=车）：
- 必须是单通道（灰度）图像：不能是RGB三通道，否则PyTorch/TensorFlow会误读为3类输入
- 数据类型必须是整型（uint8或int32）：浮点型标签（如0.0, 1.0）会导致CrossEntropyLoss计算出错
- 像素值只能是离散类别ID（0, 1, 2, ..., N-1）：不能是连续灰度值（如0–255任意数），也不支持负数
- 分辨率必须与原始图像完全一致：宽高像素数逐一对齐，缩放/裁剪必须同步进行
常用标注工具输出如何转成标准标签
LabelMe、CVAT、SuperAnnotate等工具导出的标注，通常需后处理才能用于训练：
-
LabelMe（JSON → PNG）：用官方
labelme_json_to_dataset脚本，或手动读取多边形顶点，用cv2.fillPoly绘制到全零掩膜上，再保存为uint8 - CVAT（XML/JSON → mask）：推荐导出为“Segmentation mask (PNG)”格式，检查生成的PNG是否为单通道、无调色板；若带调色板，用
img.convert('L')转灰度并映射ID -
手绘/PS生成的彩色mask：切忌直接用RGB值当类别！应建立颜色→ID映射表（如[255,0,0]→1），遍历像素查表赋值，最后转
np.uint8
数据加载时的关键预处理逻辑
在Dataset的__getitem__中，务必按顺序执行以下操作：
立即学习“Python免费学习笔记（深入）”；
- 用
PIL.Image.open(img_path)和PIL.Image.open(mask_path)读取，确保mask返回的是L模式（单通道） - 对mask做
np.array(mask, dtype=np.int64)，显式转为整型（PIL读PNG有时默认为uint8，但PyTorch要求long型标签） - 如有忽略类别（如边缘模糊区），统一设为
-1，并在Loss中设置ignore_index=-1 - 图像和mask必须使用**同一组随机变换参数**（如相同的crop坐标、flip方向），可用
torchvision.transforms.RandomApply配合自定义函数实现
验证标签质量的3个快速检查点
训练前花2分钟确认，能避开80%的“模型不收敛”假问题：
-
打印唯一值：
print(np.unique(np.array(mask)))—— 应只出现你定义的类别ID（如[0,1,2]），不含255、-1等意外值 -
检查形状和类型：
mask.shape == img.shape[:2]且mask.dtype in [np.uint8, np.int64] -
可视化叠加：用
plt.imshow(img); plt.imshow(mask, alpha=0.3, cmap='jet')看类别区域是否贴合物体边界
基本上就这些。格式看着简单，但漏掉一个dtype或错一次resize，模型就默默学错——宁可多写两行检查，别信“应该没问题”。
