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) # 合并