Jetson Nano 编译opencv-python和opencv-python-headless

在jetson nano部署剪枝版的yolov3项目,
https://github.com/Lam1360/YOLOv3-model-pruning/
需要安装albumentations
pip install albumentations会报错,opencv-python和opencv-python-headless无法安装。


该问题原因为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")
t修改为

build_headless = True2
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端剪枝后的速度没有吸引力。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章