将gray图片转成三通道 |
您所在的位置:网站首页 › 灰度图转化为三通道图 › 将gray图片转成三通道 |
在使用yolov3进行目标检测的过程中,我遇到了这样的一个奇怪的错误 果不其然代码正常运行了,detect检测出了目标,检测结束之后检查output结果让我大吃一惊 将一个通道的灰度图像,复制两次,得到三通道所需图像 写一个脚本,专门用来处理送进检测网络的图片 import cv2 import os from PIL import Image import numpy as np file_dir = "/home/jx/PyTorch-YOLOv3/data/samples/" out_dir = "/home/jx/PyTorch-YOLOv3/data/test/" a = os.listdir(file_dir) for i in a: print(i) I = Image.open(file_dir + i) L = I.convert('L') a = np.array(L) image = np.expand_dims(a, axis=2) image = np.concatenate((image, image, image), axis=-1) cv2.imwrite(out_dir + i, image)处理完成之后,再次将图片送到检测网络,继续使用 img = transforms.ToTensor()(Image.open(img_path))程序没有报错,同时出现bbox网络检测出缺陷 |
今日新闻 |
推荐新闻 |
CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3 |