基于yolov5n6和tkinker实现的检测模型的可视化界面(代码片段)

刘润森! 刘润森!     2022-11-29     288

关键词:

项目介绍

本项目基于 yolov5n6 和tkinker实现的检测模型的可视化界面

项目连接; https://github.com/MaoliRUNsen/yolov5gui

环境:

硬件: 本人电脑的显卡是 RTX 3060 ,并配置Pytorch-GPU

关于 pytorch 安装查看官方文档 PyTorch Get Started docs

软件: Pycharm 和相关的Python包

# GUI
Tkinter
# OpenCV
opencv-contrib-python>=4.6.0.66
# PyTorch + GPU CUDA
torch==1.11.0+cu115 torchvision==0.12.0+cu115 torchaudio==0.11.0 -f https://download.pytorch.org/whl/torch_stable.html

项目安装依赖

pip install -r requirements.txt  # install

整个项目的 模型下载
TensorRT, TensorFlow Edge TPU and OpenVINO Export and Inference:

weights -|- yolov5n6.pt
         |- yolov5m6.pt

项目功能

  1. 实现图像正常目标检测
  2. 实现视频正常目标检测
  3. 实现摄像头正常目标检测
  4. 实现对目标检测的画面和视频进行保存

项目演示

项目运行:

配置好相关环境和依赖,可以执行main.py

python main.py

点击yolov5,添加yolov5m6.pt模型,点击模型加载

在编辑中添加图片或者视频

图片检测检测结果;

如果在检测前,点击了图片保存

那么图像检测的结果在output文件夹中进行保存

同样,可以上传视频进行检测,如果想保存视频需要在上传前,点击保存按钮

如果遇到,图片视频上传检测不了,请麻烦将中文路径进行去除

点击摄像头,将对电脑的摄像头进行目标检测

如何自定义检测

在 weights文件夹中添加自定义检测训练的pt模型。 模型需要通过 yolov5n6.pt等进行预训练

在detect.py 中的 set_modul函数更换模型和数据训练的yaml配置文件

项目链接; https://github.com/MaoliRUNsen/yolov5gui

基于yolov5n6和tkinker实现的检测模型的可视化界面(代码片段)

项目介绍本项目基于yolov5n6和tkinker实现的检测模型的可视化界面项目连接;https://github.com/MaoliRUNsen/yolov5gui环境:硬件:本人电脑的显卡是RTX3060,并配置Pytorch-GPU关于pytorch安装查看官方文档PyTorchGetStarteddocs软件:Pych... 查看详情

基于yolo的手部检测和计数实现(课程设计,训练测试+模型剪枝+模型压缩)

用YOLOv3模型在一个开源的人手检测数据集上做人手检测,并在此基础上做模型剪枝。对于该数据集,对YOLOv3进行channelpruning之后,模型的参数量、模型大小减少80%,FLOPs降低70%,前向推断的速度可以达到原来的200%,同时可以保持m... 查看详情

基于yolov4的目标检测系统(附matlab代码+gui实现)(代码片段)

本文介绍了一种MATLAB实现的目标检测系统代码,采用YOLOv4检测网络作为核心模型,用于训练和检测各种任务下的目标,并在GUI界面中对各种目标检测结果可视化。文章详细介绍了YOLOv4的实现过程,包括算法原理、MATLAB实现代码、... 查看详情

基于深度学习的口罩检测系统(python+清新界面+数据集)(代码片段)

口罩检测系统用于日常生活中检测行人是否规范佩戴口罩,利用深度学习算法可实现图片、视频、连接摄像头等方式的口罩检测,另外支持和结果可视化。在介绍算法原理的同时,给出Python的实现代码以及PyQt的UI界面。口罩检测... 查看详情

opencv基于残差网络实现人脸检测(代码片段)

OpenCV基于残差网络实现人脸检测OpenCV3.3版本第一次把深度神经网络(DNN)模块引入到正式发布版本中,最新的OpenCV3.4中DNN模块发布了两个必杀技,一个支持FasterR-CNN的对象检测,比SSD与YOLO这些模型有更好的检测精度与小对象检测能... 查看详情

opencv+openvino实现人脸landmarks实时检测(代码片段)

...块之后,OpenCV对DNN模块支持最好的表现之一就是开始支持基于深度学习人脸检测,OpenCV本身提供了两个模型分别是基于Caffe与Tensorflow的,Caffe版本的模型是半精度16位的,tensorflow版本的模型是8位量化的。同时OpenCV通过与OpenVINOIE... 查看详情

fpga教程案例91机器视觉2——通过fpga实现基于肤色模型的人脸检测,使用matlab辅助测试

FPGA教程目录MATLAB教程目录目录1.软件版本2.基于肤色模型的人脸检测原理3.Verilog实现 查看详情

fpga教程案例90机器视觉1——通过fpga实现基于颜色模型的交通灯检测,使用matlab辅助测试

FPGA教程目录MATLAB教程目录目录1.软件版本2.基于颜色模型的交通灯检测原理3.Verilog实现 查看详情

fpga教程案例90机器视觉1——通过fpga实现基于颜色模型的交通灯检测,使用matlab辅助测试

FPGA教程目录MATLAB教程目录目录1.软件版本2.基于颜色模型的交通灯检测原理3.Verilog实现 查看详情

opencv进阶-基于coco数据集训练好的模型,修改类别显示代码,实现自定义检测物体

coco 查看详情

opencv进阶-基于coco数据集训练好的模型,修改类别显示代码,实现自定义检测物体

coco 查看详情

android实现戴口罩人脸检测和戴口罩识别(附android源码)(代码片段)

...口罩识别(附Android源码)1.戴口罩识别的方法(1)基于多类别目标检测的戴口罩识别方法(2)基于人脸检测+戴口罩分类识别方法2.戴口罩人脸数据集3.戴口罩人脸检测4.戴口罩识别模型训练5.戴口罩识别模型Android... 查看详情

毕业设计之-基于cnn实现谣言检测(代码片段)

...;这里是丹成学长,今天向大家介绍一个深度学习项目基于CNN实现谣言检测大家可用于毕业设计1.1背景社交媒体的发展在加速信息传播的同时,也带来了虚假谣言信息的泛滥,往往会引发诸多不安定因素,并对经济... 查看详情

基于yolov4的口罩检测的系统研究和实现(代码片段)

目录一、实现效果1.1、用户提交图片,检测是否带口罩 1.2、实时检测1.3、数据集介绍二、算法原理2.1数据增强:Mosaic2.2网路模型2.2.1SPP2.2.2ModifiedPAN2.2.3完整结构三、实验过程3.1推理阶段3.1.1读取并加载权重文件3.1.2读取图... 查看详情

android实现戴口罩人脸检测和戴口罩识别(附android源码)(代码片段)

...口罩识别(附Android源码)1.戴口罩识别的方法(1)基于多类别目标检测的戴口罩识别方法(2)基于人脸检测+戴口罩分类识别方法2.戴口罩人脸数据集3.戴口罩人脸检测4.戴口罩识别模型训练5.戴口罩识别模型Android... 查看详情

android实现戴口罩人脸检测和戴口罩识别(附android源码)(代码片段)

...口罩识别(附Android源码)1.戴口罩识别的方法(1)基于多类别目标检测的戴口罩识别方法(2)基于人脸检测+戴口罩分类识别方法2.戴口罩人脸数据集3.戴口罩人脸检测4.戴口罩识别模型训练5.戴口罩识别模型Android... 查看详情

基于python的智能视频分析之人数统计的多种实现(代码片段)

目录摘要11理论基础21.1传统的人群计数算法[1]31.1.1基于检测的方法31.1.2基于轨迹聚类的方法31.1.3基于回归的方法31.2基于深度学习的人群计数算法41.2.1YOLOv3目标检测算法[2]41.2.2MCNN人群计数算法[3]51.2.3LSC-CNN人群计数算法[4]51.2.4视觉... 查看详情

关于车牌识别的工作原理和技术参数

...分的训练样本见我上传的资源。实现方法(一)检测车辆(基于混合高斯模型的背景差分法)读取视频(avi格式),选定目标区域框用于检测车辆检测车辆,基于混合高斯模型的背景差分法检测前景,经过腐蚀、膨胀后,若像素值... 查看详情