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:
            ...
posted @ 2019-07-24 16:24  litchi99  阅读(243)  评论(0编辑  收藏  举报