rknn(2) example yolov5
2023-04-23 16:26 作者:FineClassFuture | 我要投稿
前面直接运行测试.rknn完成。
现在进一步:生成.rknn并pc仿真运行测试
步骤:examples/pytorch/yolov5/README.md
git clone https://github.com/ultralytics/yolov5
git checkout c5360f6e7009eb4d05f14d1cc9dae0963e949213
需要修改 yolov5/models/yolo.py文件的class Detect(nn.Module) 类的子函数forward
pip3 install pandas(2.0.0太高了,我选择==1.0.0,来降低依赖的numpy版本)
注意rknn-toolkit 1.7.3 has requirement numpy==1.19.5
python export.py --weights yolov5s.pt --img 640 --batch 1 --include torchscript、
ln -s yolov5/yolov5s.torchscript.pt yolov5s.torchscript.pt
修改test.py
仿真使用ret = rknn.init_runtime()
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