在jetson nano部署剪枝版的yolov3项目，
https://github.com/Lam1360/YOLOv3-model-pruning/
需要安装albumentations
pip install albumentations会报错，opencv-python和opencv-python-headless无法安装。
image.png
该问题原因为arm架构下，没有对应版本的opencv-python和opencv-python-headless
解决办法为使用编译。
1.git clone https://github.com/skvark/opencv-python.git
2.cd 进入，编译opencv-python
sudo python3 setup.py install
3.设置setup.py
vim setup.py
将build_headless = get_build_env_var_by_name("headless")
修改为
build_headless = True
4.编译opencv-python-headless
sudo python3 setup.py install
二者编译成功后，albumentations便可通过pip安装。
由于服务器docker的pytorch为1.6版本，jetson内docker的pytorch版本为1.4.服务器的高版本pytorch保存的pth文件在jetson低版本pytorch无法读取，所以除了项目官方给的权重外均无法测试。目前看，剪枝后的模型在服务器端减速速度提升明显，但在jetson端剪枝后的速度没有吸引力。
image.png
网友评论