使用imsave保存掩码数组时触发ValueError：找不到适配iomode `wi`的后端

报错代码片段 #

for i in range(masks.shape[0]):
    imsave(os.path.join(PATH_TO_SAVE, f"prediction_{(i+1):04d}"), masks[i])

错误信息 #

ValueError: Could not find a backend to open my_path with iomode wi

问题原因 #

• 文件名缺失图像格式后缀：生成的文件名（如prediction_0001）没有.png/.jpg这类格式标识，imsave无法自动推断要使用的图像编码后端，导致无法执行写入操作。
• 目录权限或存在性问题：PATH_TO_SAVE对应的目录不存在，或者当前运行代码的用户没有该目录的写入权限，也会触发后端找不到可用处理方式的报错。

解决方案 #

• 添加文件名格式后缀
  修改代码为文件名补充图像格式后缀（以PNG格式为例）：

  for i in range(masks.shape[0]):
      imsave(os.path.join(PATH_TO_SAVE, f"prediction_{(i+1):04d}.png"), masks[i])
  

  这样imsave能明确调用对应格式的后端完成保存。

• 确保目标目录可用
  先检查并创建目标目录（避免目录不存在的问题）：

  import os
  os.makedirs(PATH_TO_SAVE, exist_ok=True)
  

  同时确认运行代码的用户拥有该目录的写入权限。

• 显式指定图像格式（可选）
  若不想修改文件名，可通过format参数直接指定保存格式：

  for i in range(masks.shape[0]):
      imsave(os.path.join(PATH_TO_SAVE, f"prediction_{(i+1):04d}"), masks[i], format='png')
  

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