在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
⚠️ 注意事项：
