tensorflow SSD输入图像指定为三通道时的解决办法
彩色图像转灰度图像的python-opencv操作
1. RGB to GRAY
gray=cv2.cvtColor(frame,cv2.COLOR_BGR2GRAY)
2. 单通道灰度图像合成三通道灰度图像
由于TensorFlow 的目标检测SSD模型限定输入的图像需为三通道(24bit)的图像,因此,当输入视频流或图像是单通道(8bit)时,需要对其进行处理:
frame = cv2.merge([gray,gray,gray])
3. 读取文件夹下所有文件:
for root,dirs,files in os.walk(file_dir):
for file_name in files:
...