Albumentations设置is_check_shapes=False仍报尺寸匹配错误求助

可行的解决步骤： #

• 调用变换时显式传入is_check_shapes=False
  即便在A.Compose初始化时设置了该参数，部分场景下仍需在调用时显式传递，覆盖全局检查逻辑：

  transformed = img_trans(image=image, mask=mask, is_check_shapes=False)
  

• 确认Mask的维度格式
  确保你的Mask是纯二维数组（(360, 640)），而非带单通道的三维数组（(360, 640, 1)）。如果是后者，用squeeze压缩冗余维度：

  import numpy as np
  mask = np.squeeze(mask)
  

• 升级Albumentations版本
  若使用的是0.5.x及更早版本，is_check_shapes并非A.Compose的初始化参数，需升级到1.0+版本才能支持全局关闭形状检查。执行以下命令升级：

  pip install --upgrade albumentations
  

• 排查变换链中的特殊变换
  如果自定义或使用了特殊变换，检查该变换内部是否存在独立的形状校验逻辑，必要时修改或禁用该局部检查。

内容的提问来源于stack exchange，提问作者sccomp