jiangdongdizizai的博客

原创 YOLOV5 6.1 三步将第一层卷积输入从三通道改为一通道

在utils/datasets.py中读入图片使用cv2.imread(f)，此处默认将图片以bgr格式打开，一开始在这里加入IMREAD_GRAYSCALE可以改灰度图，但是后续的转hsv空间就无法进行了，因此这里不改动。由于实验的图片是黑白相机拍摄，是灰度图，在训练的时候看到第一层默认的输入通道是三，想着把三通道改成一通道会更好一些。第二步，思路是bgr格式读入图片，转hsv空间后，再转为灰度图，此处该格式，同时由于灰度图维度默认是h,w，因此扩充维度至h,w,c。