yolov5-6.1（单通道图像训练，模型转换）_yolov5单通道

作者：很楠不爱3 | 2024-03-23 01:39:00

踩

yolov5单通道

一、需要更改的内容

1、train.py文件，把ch=3改为ch=1
在这里插入图片描述
2、utils/datasets.py文件下
2.1、cv2.imread(path)改成cv2.imread(path,0)，这样子读取的图像就是单通道

在数据加载模块，把三通道改成1通道，transpose函数改为reshape，如下：

2.2、去掉红框里的代码img.shape[2]
在这里插入图片描述

3、models/yolo.py下ch=3改为ch=1
在这里插入图片描述
2.3、注释掉这行代码（三通道图像的hsv 数据增强）

4、models/common.py下,在框住的代码下，加上如下代码

if im.mode != 'L':
    im = im.convert('L')
1
2

在这里插入图片描述

二、报错问题解决

1、RuntimeError: Given groups=1, weight of size [32, 1, 6, 6], expected input[8, 3, 640, 640] to have 1 channels
解决方法：图像读取，读的是三通道rgb图像，更改imread

2、RuntimeError: Given groups=1, weight of size [32, 3, 6, 6], expected input[8, 1, 640, 640] to have 1 channels
解决方法：在train.py代码里，–cfg上，加入模型配置文件。
在这里插入图片描述
3、出现，nonetype has no attribute python_exit_status
解决方法：train.py代码里配置文件，works改为0

4、出现KMP_DUPLICATE_LIB_OK='True’的警告打印文件
解决方法：在代码运行前的位置，添加 os.environ[‘KMP_DUPLICATE_LIB_OK’]=‘True’

三、模型转换

pt模型转onnx，只需要在export.py 第457行，改channel为1
在这里插入图片描述

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/很楠不爱3/article/detail/292269