yolov8 多通道图片训练-已完成

本文详细描述了如何将Ultralytics的yolov8模型从训练RGB图像改为处理8通道图像,包括修改数据读取、配置文件、数据增强模块和相关错误处理。

最新:修改过程详细描述及完整代码更新于https://github.com/satori-hcy/yolov8_8ch

-------------------------------------------------------以下为原始草稿内容-----------------------------------

yolov8源码为训练三通道的rgb图像,本文记录修改yolov8为训练8通道的图像数据。

参考网址:https://github.com/ultralytics/ultralytics/issues/3432

总体思路:1.解决数据读取问题;2.修改网络结构的配置文件;3.解决数据增强相关错误;

1.在ultralytics/data/base.py 文件中的

def load_image(self, i, rect_mode=True)函数中修改,使im是读取到的多通道图片。

2.找到模型使用的yaml文件,添加输入通道参数:ch。我使用的是

model = YOLO("yolov8n.yaml")语句来创建了模型所以对应的是ultralytics/cfg/models/v8/yolov8.yaml文件

添加参数ch:8  (这里要几个通道就写几)

<
评论 7
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值