yolov5_for_rknn:PyTorch中的YOLOv5> ONNX> RKNN

上传者: 42116847 | 上传时间: 2021-05-26 11:23:10 | 文件大小: 6.66MB | 文件类型: ZIP
原版仓库: 修改版 yolov5 使用方法 环境要求:python version >= 3.6 模型训练:python3 train.py 模型导出:python3 models/export.py --weights "xxx.pt" 转换rknn:python3 onnx_to_rknn.py 模型推理:python3 rknn_detect_yolov5.py 注意事项:如果训练尺寸不是640那么,anchors会自动聚类重新生成,生成的结果在训练时打印在控制台,或者通过动态查看torch模型类属性获取,如果anchors不对应那么结果就会出现问题。 建议:在训练时如果size不是640,那么可以先通过聚类得到anchors并将新的anchors写入到模型配置文件中,然后再训练,防止动态获取的anchors在rknn上预测不准的问题。训练参数别忘记加上 --noautoanchor

文件下载

资源详情

[{"title":"( 120 个子文件 6.66MB ) yolov5_for_rknn:PyTorch中的YOLOv5> ONNX> RKNN","children":[{"title":".dockerignore <span style='color:#111;'> 3.53KB </span>","children":null,"spread":false},{"title":".gitattributes <span style='color:#111;'> 75B </span>","children":null,"spread":false},{"title":"train.py <span style='color:#111;'> 30.82KB </span>","children":null,"spread":false},{"title":"Dockerfile <span style='color:#111;'> 821B </span>","children":null,"spread":false},{"title":"additional_requirements.txt <span style='color:#111;'> 105B </span>","children":null,"spread":false},{"title":"......","children":null,"spread":false},{"title":"<span style='color:steelblue;'>文件过多,未全部展示</span>","children":null,"spread":false}],"spread":true}]

评论信息

  • ma_cainiao_ming :
    很好,直接放到开发板上就能跑起来
    2021-11-19

免责申明

【只为小站】的资源来自网友分享,仅供学习研究,请务必在下载后24小时内给予删除,不得用于其他任何用途,否则后果自负。基于互联网的特殊性,【只为小站】 无法对用户传输的作品、信息、内容的权属或合法性、合规性、真实性、科学性、完整权、有效性等进行实质审查;无论 【只为小站】 经营者是否已进行审查,用户均应自行承担因其传输的作品、信息、内容而可能或已经产生的侵权或权属纠纷等法律责任。
本站所有资源不代表本站的观点或立场,基于网友分享,根据中国法律《信息网络传播权保护条例》第二十二条之规定,若资源存在侵权或相关问题请联系本站客服人员,zhiweidada#qq.com,请把#换成@,本站将给予最大的支持与配合,做到及时反馈和处理。关于更多版权及免责申明参见 版权及免责申明