yolov4-tiny训练自己的模型并部署到RK的NPU上推理
文章目录前言一、安装darknet训练框架二、模型转换二、RKNN推理前言使用yolov4-tiny训练西瓜检测模型,部署到RKNPU上推理Python单线程速度35FPS以上一、安装darknet训练框架本人使用得是这里的darknetReadme里有详细写明数据准备,cfg修改,训练步骤等,本文不在赘述。二、模型转换训练完成后得到yolov4-yiny.cfg和yolov4-tiny.weight, 转换为yolov4-tiny.rknnfrom PIL import Imagei
原创
2020-10-17 16:53:05 ·
7500 阅读 ·
12 评论