在 u -net 等医学图像分割任务中，必须保证图像与对应掩码经历 ** 完全一致的随机几何变换 **（如旋转、翻转），否则标签错位将导致模型学习失效；直接对二者独立调用 `transform` 会生成不同随机参数，正确做法是通道拼接后统一变换再切分。
问题根源在于：torchvision.transforms.v2（以及 v1）中的 RandomRotation、RandomHorizontalFlip 等随机变换操作每次调用都会 重新采样随机参数 （例如不同的旋转角度或是否翻转）。当你分别对 image 和 mass_mask 调用 self.transform() 时，它们各自触发了独立的随机过程——图像可能被顺时针旋转 12.7°并水平翻转，而掩码却被逆时针旋转 5.3°且未翻转，最终导致空间错配，如你提供的图示所示。
✅ 正确解法：将图像与掩码沿通道维度（dim=0）拼接为单个张量，一次性应用变换，再按通道数切分还原。这确保了所有空间操作（旋转中心、插值方式、翻转决策等）完全同步。
以下是修正后的 __getitem__ 关键代码段（适配 torchvision.transforms.v2）：
def __getitem__(self, index): dict_path = os.path.join(self.dict_dir, self.data[index]) patient_dict = torch.load(dict_path) image = patient_dict['image'].unsqueeze(0) # shape: [1, H, W] mass_mask = patient_dict['mass_mask'].unsqueeze(0) # shape: [1, H, W] mass_mask = torch.clamp(mass_mask, 0.0, 1.0) # 更安全的二值化替代原逻辑 if self.transform is not None: # ✅ 拼接：[1, H, W] + [1, H, W] → [2, H, W] combined = torch.cat([image, mass_mask], dim=0) # ✅ 统一变换（所有像素级 / 几何操作共享同一随机种子）transformed = self.transform(combined) # ✅ 切分还原 image = transformed[0:1] # 取第 0 个通道 → [1, H, W] mass_mask = transformed[1:2] # 取第 1 个通道 → [1, H, W] return image, mass_mask⚠️ 注意事项：
- transform 必须支持多通道输入：torchvision.transforms.v2.RandomRotation 等默认支持任意通道数，无需额外配置；但若使用自定义 Lambda 或旧版 v1，需确认其兼容性。
- 插值模式匹配 ：对掩码应使用 最近邻插值（interpolation=InterpolationMode.NEAREST） 避免引入灰度值（如旋转后出现 0.3 这样的非 0 / 1 值）。推荐显式声明：
train_transform = T.Compose([T.RandomRotation(degrees=35, expand=True, fill={0: 0.0, 1: 0.0}), # 图像填 255→归一化后≈1.0；掩码填 0 T.RandomHorizontalFlip(p=0.5), T.RandomVerticalFlip(p=0.5), # 若需归一化，应在最后单独加 T.Normalize，避免影响掩码 ]) - fill 参数需区分对待：RandomRotation 的 fill 默认填充 0，但图像常需填背景值（如 255→归一化后为 1.0），而掩码应填 0。使用字典形式 fill={0: 1.0, 1: 0.0} 可精确指定第 0 通道（图像）和第 1 通道（掩码）的填充值。
- 避免归一化污染掩码：切勿在 Compose 中对拼接张量使用 T.Normalize——它会错误地将掩码值（0/1）标准化。归一化应仅作用于图像通道，建议拆分为两步：先做几何变换（拼接处理），再对图像单独归一化。
? 进阶提示：若需更灵活控制（如弹性形变、亮度调整仅作用于图像），可考虑 albumentations 库，它原生支持 image + mask 同步变换（通过 albumentations.Compose(…, additional_targets={‘mask’: ‘mask’})），语义更清晰，适合复杂 pipeline。
综上，通道拼接法是 PyTorch 生态下最轻量、可靠且无需引入新依赖的解决方案，能从根本上杜绝图像 - 掩码变换失步问题，是医学图像分割数据加载器的标准实践之一。
