logo资料库

海思Hi35xx系列运行YoloV3以及YoloV3-tiny指南.pdf

第1页 / 共4页
第2页 / 共4页
第3页 / 共4页
第4页 / 共4页
资料共4页,全文预览结束
海思 Hi35xx 系列 运行 YoloV3 以及 YoloV3-tiny 指南 一、YoloV3 a) 训练模型 参考链接: https://blog.csdn.net/qq_21578849/article/details/84980298 b) darknet 转 caffe https://github.com/ChenYingpeng/caffe-yolov3 https://github.com/ChenYingpeng/darknet2caffe 在此感谢 ChenYingpeng 大神 要验证在PC 上通过caffe 框架测试caffe 模型可以正常检测目标, 并且准确率达标。 c) caffe 转 wk 参考海思 SDK 中的 SVP 例程。 建议首先阅读文档《HiSVP 开发指南.pdf》、《HiSVP API 参 考.pdf》、《HiSVP 量化库使用指南.pdf》 说明:步骤 b 中生成的 Prototxt 可以直接用,ruyi studio 会自 动进行规范化。 d) NNIE 部署 参考“会飞的牛”的 Github https://github.com/yangfei963158659/nnie 非常详细,直接可用。在此感谢。 二、YoloV3-Tiny a) 修改 yolov3-tiny 模型 cfg 文件,去掉最后一个 stride=1 的
maxpool 层 b) 训练过程和 YoloV3 的训练差不多 预训练模型的生成,参考: https://blog.csdn.net/qq_36302589/article/details/85222625 c) darknet 转 caffe(同 YoloV3) d) caffe 转 wk(同 YoloV3) e) NNIE 部署 同 YoloV3,需要修改几个地方,请随时关注会飞的牛的 github 上 的 repo 更新。 根据情况需要修改 1. 原来 3 个 yolo 层变成获取两个 yolo 层的 output
2. anchor 修改 3. 类别数按实际情况修改
三、运行时间参考 a) YoloV3 416*416 标准版在 3519A 上的推理时间为 85ms 左右。 b) YoloV3-tiny 416*416 模型在 3519A 上的推理时间为 10ms 左右。 四、技术支持(Free) 欢迎加入海思 AI 交流 QQ 群 群号:202924426 本指南作者 Email:xiwei.sdu@gmail.com
分享到:
收藏