报错如下:
原因:灰度图像为单通道图像
解决方法:将单通道图像转换为三通道图像,再转换为coco数据集用于训练
单通道转三通道图像:
import cv2
import os
import numpy as np
path = 'D:/ykWork/dlImg/gray/input' # 源文件所在目录 图片文件
savefilepath = 'D:/ykWork/dlImg/gray/' # 输出文件所在目录 图片文件
datanames = os.listdir(path)
for i in datanames:
img = cv2.imread(path + '/' + str(i))
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
img2 = np.zeros_like(img)
img2[:, :, 0] = gray
img2[:, :, 1] = gray
img2[:, :, 2] = gray
cv2.imwrite(savefilepath + i, img2)