欢迎光临散文网 会员登陆 & 注册

rknn(2) example yolov5

2023-04-23 16:26 作者:FineClassFuture  | 我要投稿

    前面直接运行测试.rknn完成。

    现在进一步:生成.rknn并pc仿真运行测试

    步骤:examples/pytorch/yolov5/README.md

  1.  git clone https://github.com/ultralytics/yolov5

  2. git checkout c5360f6e7009eb4d05f14d1cc9dae0963e949213

  3. 需要修改 yolov5/models/yolo.py文件的class Detect(nn.Module) 类的子函数forward

  4.  pip3 install pandas(2.0.0太高了,我选择==1.0.0,来降低依赖的numpy版本)

    注意rknn-toolkit 1.7.3 has requirement numpy==1.19.5

  5. python export.py --weights yolov5s.pt --img 640 --batch 1 --include torchscript、

  6. ln -s yolov5/yolov5s.torchscript.pt yolov5s.torchscript.pt

  7. 修改test.py 

    仿真使用ret = rknn.init_runtime()

  8.  python3 test.py

    现在特意看了下系统监视器,发现只有一个cpu核 100%


同样yolov5s,

examples/onnx/yolov5/yolov5.onnx => yolov5.rknn 7.4MB

examples/pytorch/yolov5/yolov5s.torchscript.pt => yolov5s.rknn 14.6MB


rknn(2) example yolov5的评论 (共 条)

分享到微博请遵守国家法律