- 博客(2)
- 收藏
- 关注
RSS订阅原创 YOLOv8 pt模型转化为onnx格式
本文介绍了将YOLOv8训练好的pt模型转换为ONNX格式的方法。主要内容包括：1）YOLOv8模型训练过程，包括数据集配置和训练参数设置；2）修改模型代码head.py和model.py文件，调整forward函数和初始化方法以支持ONNX导出；3）创建转换脚本pt-onnx.py，加载训练好的pt模型和配置文件，最终导出为包含6个输出层的ONNX格式模型。该转换过程为后续将模型转换为RKNN格式提供了必要的中间步骤。
2026-01-02 15:32:23
491
原创 YOLOv5改进系列——融合BiFPN（双向特征金字塔网络）增加P2微小目标检测层
本文探讨了改进YOLOv5进行小目标检测的方法，重点关注BiFPN（双向特征金字塔网络）的融合应用。主要内容包括：1）BiFPN的核心优势，如双向特征融合机制和加权融合策略；2）小目标检测的定义及技术难点；3）两种YOLOv5融合方案：基于BiFPN_Add2的简化版本和结合BiFPN_Add3的增强版本，均通过添加P2检测层提升微小目标（8-32像素）识别能力。实验表明，改进后的模型参数量增幅可控，检测性能尤其是小目标mAP值得到提升。该研究为小目标检测任务提供了有效的网络结构优化思路。
2025-07-04 23:40:02
2251
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人