Unet分割自己数据集时出现expected input[1, 4, 384, 512] to have 3 channels, but got 4 channels instead

原因:png格式问题

import os
import cv2
'''png格式常常是32位的RGBA格式,A代表透明度,
   光是更改图片后缀,不能改变图片的位数,
   需要在openCV中进行色彩空间的转换,
   将png格式的32位RGBA转为png格式的24位RGB'''

def convert2jpg(filename):                                      # 将彩色图转灰度图的函数
    img = cv2.imread(filename, 1)                 # 1是以彩色图方式去读
    jpg_img = cv2.cvtColor(img, cv2.COLOR_BGRA2BGR)
    cv2.imwrite('0081.png', jpg_img)             # 保存在新文件夹下,且图名中加GRAY
    
                         # 设置为新目录为输出文件夹

                          # 遍历输入路径,得到图片名
filename = '008.png'
convert2jpg(filename)

你可能感兴趣的:(图像分割,opencv)