RK3399平台开发系列讲解(系统篇)1.23、新yolo算法在rk平台上的挣扎历程

本文档详细记录了在RK3399平台上部署新版YOLO算法的过程,包括环境配置、模型转换、版本匹配、模型重建及后处理等挑战。在Pytorch 1.6到1.2的版本适配中,作者解决了AMP、tensorboard、Hardswish等依赖问题,并简化了模型结构。经过多次调试和优化,最终实现了模型的正确运行和推理,为平台上的AI算法部署提供了参考。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

第一章文档介绍

本文档记录总结了新版yolo算法在RK平台上部署过程。主要解决的问题在于版本的匹配,其次是网络的重建,最后是输出的处理。其中RK平台目前最新toolkit是1.4.1,对于本人熟悉的pytorch深度学习框架仅支持1.2版本(目前最新为1.7)。直接套用原始的代码在实验中出现了结果完全无法对应的情况(我愿称之为:玄学),因此对网络的重建以及原子操作的拆分很重要,由此确保结果的一致。最后,RK平台的工具只支持4维向量的操作,但yolo算法最后一层为5维向量,无法直接完成输出转换,由此需要后期自行处理。
本文是我一步一步完全根据代码调试的过程进行,期间耗时较长,并且后续还有很多需要调整的地方,目的是能将比较新的算法应用的RK设备上。本文图片较多,大多是官方源码,但是由于其代码量实在巨大,无法面面俱到,只对一些出现异常的地方进行了解释。篇幅较长,渐觉枯燥,因此后文笔风逐渐放浪,不喜勿喷~
目的在于为读者提供经验与项目参考。

第二章项目介绍

针对各AI算法(如车牌识别)的平台部署。

第三章素材处理

3.1.素材采集

评论 9
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

内核笔记

你的鼓励是我的动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值