作者:殷忆枫

RV1126笔记十九:吸烟行为检测及部署<六>

若该文为原创文章,转载请注明原文出处。

模型测试

一、pt文件测试

pt文件是在windows下训练生成的,测试环境为py3.8

1、首先查看虚拟环境列表,然后切换于是py3.8的虚拟环境。

conda env list          // 查看虚拟环境列表
conda activate yolo     // 切换虚拟环境

2、进入yolov5-master_rk目录

yolov5前面已经下载过了,下载的是rk优化后的yolov5版本,或是用yolov5-5.0版本也可以,亲测5.0版本可用。

 测试命令

python detect.py --source 0 --weights runs/train/exp4/weights/best.pt

best.pt文件需要修改相对应的路径,--source 0是电脑USB摄像头。

二、onnx模型测试

测试説明参考

lock