YOLOv5实现完手机检测
模型和数据集下载:
1、玩手机检测数据
2、YOLOv5玩手机检测模型
3、YOLOv5玩手机检测模型+pyqt界面
4、 VOC手机检测识别数据集
5、yolov7玩手机检测+训练好的玩手机的模型
6、Darknet版YOLOv3玩手机检测+weights权重+标注好的数据集
7、Darknet版YOLOv4玩手机检测+weights权重+数据集
- 其他数据集和模型可见 :YOLO系列目标检测数据集大全
配置环境
首先需要安装anaconda3和pycharm,若不知道如何安装,请参考这篇博客
安装好anaconda后,在上面新建python3.8虚拟环境,打开anaconda终端,输入如下命令:
conda create -n py38 python=3.8 # py38是虚拟环境的名字,可以更改
conda activate py38 # 激活创建好的虚拟环境
gpu版本和cpu版本的pytorch安装方法
YOLOv5项目在CPU和GPU下均可运行,但是CPU运行速度很慢,当然使用已经训练好完手机检测模型用来测试得出结果,CPU也是可行的,如果需要自己重新训练数据集,最好还是得有一个可用的英伟达的显卡。
如果有显卡,需要确定支持的CUDA的版本,所以有显卡的可以安装GPU版本的Pytorch,只有CPU的直接安装CPU版本的pytorch。在anaconda中激活创建好的虚拟环境,执行对应版本的命令就可以 (安装时间看具体网速,一般要比较久的时间):
# GPU版本pytorch安装命令
pip install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
# CPU版本pytorch安装命令
conda install pytorch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1 cpuonly -c pytorch
执行命令后如下,需要激活虚拟环境后再安装,也就是安装在py38环境中:
安装YOLOv5所需其他依赖
首先下载YOLOv5完手机检测模型;若需要pyqt界面,下载YOLOv5完手机检测模型+pyQT可视化界面。
项目文件下有需要安装的依赖要求在requirements.txt文件中,我们在pycharm终端里面来安装其他依赖。
在pycharm中打开项目文件夹下的requiremes.txt,注释掉已经安装好的包,此外还安装pyqt5用于可视化界面:
接下来打来pycharm自带终端:
输入命令:pip install -r requirements.txt
安装即可,pip install 也可以添加清华源安装加速,则输入:
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install pyqt5 -i https://pypi.tuna.tsinghua.edu.cn/simple
使用训练好的YOLov5玩手机检测模型运行推理
训练好的玩手机检测模型保存在runs/train下的文件夹中,包含YOLOv5s和YOLOv5m两种训练好的模型。
在pycharm终端中执行如下命令:
python detect.py --source play_phone-dataset/img/1.jpg --weights runs/train/play_phone-v5s/weights/best.pt
若需要在运行过程中展示图片,需要在输入命令行中加入参数--view-img
python detect.py --source play_phone-dataset/img/1.jpg --weights runs/train/play_phone-v5s/weights/best.pt --view-img
yolov5支可以检测图像、视频、批量图像检测以及调用自带摄像头或者USB摄像头,具体的使用方法如下:
python detect.py --source 0 # 调用自带摄像头
file.jpg # image
file.mp4 # video
path/ # directory目录
path/*.jpg # glob所有图片
如检测视屏,需要将如下命令中的file.mp4换成自己的视屏文件路径即可:
python detect.py --source file.mp4 --weights runs/train/play_phone-v5s/weights/best.pt --view-img
如果需要调用电脑自带摄像头检测:
python detect.py --source 0 --weights runs/train/play_phone-v5s/weights/best.pt --view-img
批量检测图片则执行如下命令:
python detect.py --source play_phone-dataset/img --weights runs/train/play_phone-v5s/weights/best.pt
到此就完成了如何使用已经训练好的玩手机检测模型来进行推理得到识别结果了,其他YOLOv5识别模型也是这种步骤来实现的。文章来源:https://www.toymoban.com/news/detail-491738.html
玩手机检测结果可视化:
文章来源地址https://www.toymoban.com/news/detail-491738.html
到了这里,关于【YOLOv5实现玩手机检测】的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!