labelme2coco转化灰度图片时报错assert image.ndim == 3 AssertionError

报错如下：

原因：灰度图像为单通道图像

解决方法：将单通道图像转换为三通道图像，再转换为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)