torch将图片批量转化为单通道 做语义分割,自己做的mask图片,存的时候虽然是全黑的但是保存之后格式仍然是3通道,,,输进网络训练就发现虽然tensor的shape是一样的,,但是数量就差了3倍 所以需要把mask转为单通道灰度图 代码如下: 代码要放在图片所在路径里 from PIL import Image import os path = os.getcwd() # 获取当前路径 file_list = os.listdir() for file in file_list: