检测阶段:
模型下载链接:https://github.com/ultralytics/yolov5/releases
开始部署环境:
出现的问题:No module named “skbuild”
解决: pip install scikit-build
测试
python3 detect.py --source data/images --weights model/yolov5l.pt --conf 0.25
ONNX和Pytorch格式文件的相互转换:
Pytorch模型导出为ONNX格式:
实验室服务器
没有python3.8版本
所以首先安装anaconda环境:
wget -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-2020.11-Linux-x86_64.sh
bash ****.sh
创建anaconda环境:
conda create -n yolov5 python==3.8
激活环境
source activate yolov5
安装onnx
pip install coremltools>=4.1 onnx>=1.8.1 scikit-learn==0.19.2 # export requirements
安装缺失的python包后运行
python models/export.py --weights model/yolov5lls.pt --img 640 --batch 1 # export at 640x640 with batch size 1
出现的问题:
1.
2.
No module named ‘PIL’
pip install image
3.
No module named cv2
pip install opencv-python
4.
No module named tqdm
pip install tqdm
5.
No module named torchvision
pip install torchvision
6.
No module named yaml
pip install pyyaml
7.
No module named matplotlib
pip install matplotlib
8.
No module named pandas
pip install pandas
9.
No module named seaborn
pip install seaborn