基于YOLOv6m的接打电话检测识别分析系统
创始人
2024-05-11 16:10:07
0次

本身在实际项目开发应用中YOLO都是目标检测任务的绝对主力,从v3一直跟着到了v7,做了很多的项目,处理了很多的数据,当然了也积累了一些自己的成果和心得,这里主要是以不常用到的yolov6m系列的模型来开发构建接打电话行为检测识别模型,首先看下效果图:

如果不会使用yolov6项目的话可以参考我之前的教程:

《基于美团技术团队最新开源的yolov6模型实现裸土检测》

教程这里就不再赘述了。

这里使用的是精度更高的yolov6m系列的模型,如下:

# YOLOv6m6 model
model = dict(type='YOLOv6m6',pretrained='weights/yolov6m6.pt',depth_multiple=0.60,width_multiple=0.75,backbone=dict(type='CSPBepBackbone_P6',num_repeats=[1, 6, 12, 18, 6, 6],out_channels=[64, 128, 256, 512, 768, 1024],csp_e=float(2)/3,fuse_P2=True,),neck=dict(type='CSPRepBiFPANNeck_P6',num_repeats=[12, 12, 12, 12, 12, 12],out_channels=[512, 256, 128, 256, 512, 1024],csp_e=float(2)/3,),head=dict(type='EffiDeHead',in_channels=[128, 256, 512, 1024],num_layers=4,anchors=1,strides=[8, 16, 32, 64],atss_warmup_epoch=4,iou_type='giou',use_dfl=True,reg_max=16, #if use_dfl is False, please set reg_max to 0distill_weight={'class': 1.0,'dfl': 1.0,},)
)solver = dict(optim='SGD',lr_scheduler='Cosine',lr0=0.0032,lrf=0.12,momentum=0.843,weight_decay=0.00036,warmup_epochs=2.0,warmup_momentum=0.5,warmup_bias_lr=0.05
)data_aug = dict(hsv_h=0.0138,hsv_s=0.664,hsv_v=0.464,degrees=0.373,translate=0.245,scale=0.898,shear=0.602,flipud=0.00856,fliplr=0.5,mosaic=1.0,mixup=0.243,
)

接下来看下数据集:

数据集种隐私敏感信息都做了脱敏处理了。

YOLO格式的标注数据如下:

实例标注内容如下:

1 0.5875 0.419444 0.165625 0.394444

VOC格式标注数据如下所示:

实例标注内容如下:

33dd213ba-4491-4425-90f8-6a7012f9b9f5.jpg3dd213ba-4491-4425-90f8-6a7012f9b9f5.jpgUnknown96054030personUnspecified1090897960132personUnspecified00222855292personUnspecified00309273139personUnspecified001903223065personUnspecified001306116797personUnspecified002145926394

默认执行100次epoch迭代计算,结果详情如下:

LABEL可视化:

混淆矩阵:

训练可视化:

F1值曲线和PR曲线:

batch检测实例:

基于界面实现可视化推理:

上传图像:

检测推理:

相关内容

热门资讯

前端-session、jwt 目录:   (1)session (2&#x...
linux入门---制作进度条 了解缓冲区 我们首先来看看下面的操作: 我们首先创建了一个文件并在这个文件里面添加了...
关于测试,我发现了哪些新大陆 关于测试 平常也只是听说过一些关于测试的术语,但并没有使用过测试工具。偶然看到编程老师...
前缀和与对数器与二分法 1. 前缀和 假设有一个数组,我们想大量频繁的去访问L到R这个区间的和,...
nodejs:本地安装nvm实... 一、背景-使用不同版本node的原因 vue3+ts、nuxt3版本,node...
JAVA集合知识整理 Java集合知识整理 HashMap相关 HashMap的底层数据结构:jdk1.8之...
无刷直流电机介绍及单片机控制实... 无刷直流电机介绍及单片机控制实例前言基本概念优势与劣势使用寿命基本结构使用单片机控制实例电子调速器&...
fwdiary(2) dp2 1.传纸条  AcWing 275. 传纸条 - AcWing 走两条路,走一条最大的...
常用的DOS命令 常用的DOS命令 DOS(Disk Operating System,磁...
<C++> 类和对象(下) 1.const成员函数将const修饰的“成员函数”称之为const成员函数,cons...