训练CTPN报错IndexError: list index out of range

IndexError: list index out of range

训练CTPN的时候有些图片报这个错。
出错代码如下。因为在减均值的时候是三个通道分别减去三个均值，某些图片是两个通道的，长度不匹配所以报错了。
解决：转换为RGB三通道图 在这

    vggMeans = [122.7717, 102.9801, 115.9465] 
    imageList = cv2.split(image.astype(np.float32)) # 分离通道
    imageList[0] = imageList[0]-vggMeans[0]  
    imageList[1] = imageList[1]-vggMeans[1]
    imageList[2] = imageList[2]-vggMeans[2]
    image = cv2.merge(imageList)  # 合并