0.目标
今天首次加入了rknn的QQ交流群,第三群,看到有人贴出了yolov5的实测速度:
这里面看到最慢的是yolov5m.我们假定它的精度最好。
1.在 rk3588上下载相关的模型
git clone https://github.com/airockchip/yolov5.git
注意,rk3588的发布目录已经改至airockchip目录
接下来生成模型:
root@lubancat:~/git/yolov5/models# sudo python3 ./yolo.py
Traceback (most recent call last):
File "/root/git/yolov5/models/./yolo.py", line 24, in <module>
from models.common import *
File "/root/git/yolov5/models/common.py", line 21, in <module>
import pandas as pd
ModuleNotFoundError: No module named 'pandas'
上面这类缺失的软件包可以使用:
python3.11 -m pip install pandas --timeout 200 -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com
在中国镜像站安装。注意那个PIL,用pillow替代:
ModuleNotFoundError: No module named 'PIL'
root@lubancat:~/git/yolov5/models# python3 -m pip PIL IPython --timeout 200 -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com
ERROR: unknown command "PIL"
root@lubancat:~/git/yolov5/models# python3 -m pip install PIL --timeout 200 -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com
Looking in indexes: http://mirrors.aliyun.com/pypi/simple/
ERROR: Could not find a version that satisfies the requirement PIL
ERROR: No matching distribution found for PIL
root@lubancat:~/git/yolov5/models# python3 -m pip install pillow --timeout 200 -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com
最终,运行yolo项目根目录的benchmarks即可:
sudo python3 ./benchmarks.py
<待续...>