基于深度学习的高精度老鼠检测识别系统（PyTorch+Pyside6+YOLOv5模型）

摘要：基于深度学习的高精度老鼠检测识别系统可用于日常生活中检测与定位老鼠目标，利用深度学习算法可实现图片、视频、摄像头等方式的老鼠目标检测识别，另外支持结果可视化与图片或视频检测结果的导出。本系统采用YOLOv5目标检测模型训练数据集，使用Pysdie6库来搭建页面展示系统，同时支持ONNX、PT等模型作为权重模型的输出。本系统支持的功能包括老鼠训练模型的导入、初始化；置信分与IOU阈值的调节、图像上传、检测、可视化结果展示、结果导出与结束检测；视频的上传、检测、可视化结果展示、结果导出与结束检测；摄像头的上传、检测、可视化结果展示与结束检测；已检测目标列表、位置信息；前向推理用时。另外本老鼠检测识别系统同时支持原始图像与检测结果图像的同时展示，原始视频与检测结果视频的同时展示。本博文提供了完整的Python代码和使用教程，适合新入门的朋友参考，完整代码资源文件请转至文末的下载链接。

基本介绍

近年来，机器学习和深度学习取得了较大的发展，深度学习方法在检测精度和速度方面与传统方法相比表现出更良好的性能。YOLOv5是单阶段目标检测算法YOLO的第五代，根据实验得出结论，其在速度与准确性能方面都有了明显提升，开源的代码可见https://github.com/ultralytics/yolov5。因此本博文利用YOLOv5检测算法实现一种高精度老鼠识别检测模型，再搭配上Pyside6库写出界面系统，完成目标检测识别页面的开发。注意到YOLO系列算法的最新进展已有YOLOv6、YOLOv7、YOLOv8等算法，将本系统中检测算法替换为最新算法的代码也将在后面发布，欢迎关注收藏。

环境搭建

（1）下载YOLOv5源码库，放到自己电脑的目录，之后打开cmd进入到YOLOv5目录里面，本文演示的目录是：D:\vscode_workspace\yolov5
（2）利用Conda创建环境（Anacodna），conda create -n yolo5 python=3.8 然后安装torch和torchvision（pip install torch1.10.0+cu113 torchvision0.11.0+cu113 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple）其中-i https://pypi.tuna.tsinghua.edu.cn/simple代表使用清华源，这行命令要求nvidia-smi显示的CUDA版本>=11.3,最后安装剩余依赖包使用：pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple