一种基于迁移学习的改进YOLOv8鱼类图像识别方法

AITNT
正文
推荐专利
一种基于迁移学习的改进YOLOv8鱼类图像识别方法
申请号:CN202411438377
申请日期:2024-10-15
公开号:CN119445348A
公开日期:2025-02-14
类型:发明专利
摘要
本发明公开了一种基于迁移学习的改进YOLOv8鱼类图像识别方法,包括以下步骤:收集鱼类图像数据集,并对数据集进行预处理;将鱼类图像数据集按照鱼的种类对应着标注上种类名称,并将其按比例划分为训练集、验证集和测试集;对标准的YOLOv8模型进行改进;采用ImageNet数据集中所有鱼类图像对改进的YOLOv8模型进行预训练,并将得到的权重参数迁移至模型中;利用训练集和验证集对迁移学习后的改进YOLOv8模型进行优化参数训练;通过优化后的改进YOLOv8模型对测试集进行目标检测,输出鱼类识别结果。本发明有效地提升了特征提取和识别的精度,同时降低了计算资源需求,增强了模型的鲁棒性和适应性;本发明显著提高了模型的学习效率和识别精度,减少了数据依赖,使模型具有较强的泛化能力。
技术关键词
图像识别方法 输出特征 卷积模块 样本 注意力机制 数据 通道 训练集 网络 冗余特征 轮廓特征 坐标 超参数 尺寸 图片 鲁棒性
添加客服微信openai178,进AITNT官方交流群
驱动智慧未来:提供一站式AI转型解决方案
沪ICP备2023015588号