1 安装

conda activate py38
git clone https://github.com/ppogg/YOLOv5-Lite
pip install -r requirements.txt

2 下载预训练的权重

预训练权重可以在官网下载，我这里下载的是v5lite-s

3 推理

图片推理

python3 ./detect.py --weights ./weights/v5lite-e.pt --source ././python_demo/openvino/bike.jpg

推理结果在runs/detect/expxx中

视频推理

python3 ./detect.py --weights ./v5lite-s.pt --source ~/path/3.mp4

推理结果是直接视频合成了框生成好的