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摄像头。