这些小活动你都参加了吗?快来围观一下吧!>>
电子产品世界 » 论坛首页 » 嵌入式开发 » STM32 » 【转载】基于机器学习的手势数字识别检测--from森

共1条 1/1 1 跳转至

【转载】基于机器学习的手势数字识别检测--from森

工程师
2025-05-19 10:58:20     打赏

 一、背景意义

       手势数字识别是人机交互领域的重要研究方向,能够有效提升智能设备的使用体验。通过手势识别,用户可以以更自然的方式与设备进行交互,尤其在智能家居、机器人和虚拟现实等应用中具有广泛的前景。准确识别手势数字(从数字0到25)不仅有助于

提升设备的智能化水平,还能够为残障人士提供更便捷的操作方式,推动社会的包容性发展。

二、数据集

2.1数据采集

       数据采集是制作手势数字识别检测数据集的第一步,旨在收集各类手势数字图像。整个过程可以分为以下几个方面:

设备选择:选择高分辨率摄像头或智能手机进行图像采集,以确保图像质量。光照条件也应考虑,最好在自然光下拍摄,以减少阴影和反光的影响。

采集环境:选择多个不同的环境进行拍摄,例如室内、户外以及不同的背景,以确保模型能够学习到在各种条件下的手势特征。

样本多样性:对于每个数字手势(从数字0到25),需要收集到多样的手势表现方式。不同的人可能在手势上存在差异,因此应邀请多位参与者进行手势展示,确保数据的多样性和丰富性。

图像数量:每个分类应尽量收集相同数量的图像,以保证数据集的均衡性。通常建议每个数字手势至少收集数百张图像,以便于后续模型训练。

       数据清洗的目的是提高数据集的质量,剔除不符合标准的样本。清洗过程主要包括:

去重处理:检查数据集中是否存在重复图像,利用图像比较工具进行去重,确保每个样本都是唯一的。

质量审核:对所有图像进行质量检查,剔除模糊、低分辨率或与手势数字识别无关的图像,以确保每张图像都能够清晰地展示手势的特征。

一致性检查:确保所有图像的格式和尺寸一致,比如将所有图像调整为640x480像素,以便后续处理和模型训练。

类别数量检查:检查每个类别的图像数量是否均衡,确保在模型训练过程中避免类别不均衡问题,以提升模型的泛化能力。

2.2数据标注

       数据标注是制作手势数字识别数据集的重要环节,主要目标是为每张图像中的手势数字添加标签。标注过程的主要步骤包括:

选择标注工具:使用标注工具(如LabelImg)进行标注,确保该工具支持导出所需格式的标注文件。

绘制边界框:逐一打开图像,使用矩形工具围绕每种手势数字(如数字0到25)绘制边界框,并标注为对应的类别。

确保标注一致性:在标注过程中,确保每个绘制的边界框准确包围目标,避免漏标或误标,以提高数据集的准确性。

保存标注结果:完成标注后,将标注结果保存为指定格式(如XML或TXT),以便后续模型训练时使用。

标注质量审查:对标注结果进行复审,确保每个手势数字均被准确标注,必要时进行二次标注,以提高标注的质量。

       使用LabelImg进行手势数字识别数据集的标注过程相对复杂且工作量较大。标注人员首先需启动LabelImg软件,并选择待标注的图像文件夹。接下来,逐一打开每张图像,使用矩形工具围绕每个手势

(如数字0至25)绘制边界框,并为每个框输入相应的类别名称。整个过程要求标注人员保持高度专注,以确保每个边界框准确地覆盖目标,避免漏标或误标。完成标注后,需将每张图像的标注结果保存,

且在标注过程中还需定期进行交叉验证以提高标注的准确性。由于涉及大量图像和细致的标注工作,这一过程既耗时又需耐心,标注人员必须具备对手势特征的深刻理解,以确保最终数据的准确性和可靠性。

1.png


手势图片数据集中包含以下几种类别:


数字0:手势表示的数字零。

数字1:手势表示的数字一。

数字2:手势表示的数字二。

数字3:手势表示的数字三。

数字4:手势表示的数字四。

数字5:手势表示的数字五。

数字6:手势表示的数字六。

数字7:手势表示的数字七。

数字8:手势表示的数字八。

数字9:手势表示的数字九。

数字10:手势表示的数字十。

数字11:手势表示的数字十一。

数字12:手势表示的数字十二。

数字13:手势表示的数字十三。

数字14:手势表示的数字十四。

数字15:手势表示的数字十五。

数字16:手势表示的数字十六。

数字17:手势表示的数字十七。

数字18:手势表示的数字十八。

数字19:手势表示的数字十九。

数字20:手势表示的数字二十。

数字21:手势表示的数字二十一。

数字22:手势表示的数字二十二。

数字23:手势表示的数字二十三。

数字24:手势表示的数字二十四。

数字25:手势表示的数字二十五。

2.3数据预处理

数据预处理是为模型训练做准备的重要步骤,主要包括以下几个方面:

数据增强:通过图像翻转、旋转、缩放和裁剪等方法对原始图像进行增强,增加数据集的多样性和复杂性,从而提高模型的鲁棒性。

归一化处理:对图像的像素值进行归一化,将其缩放到[0, 1]或[-1, 1]的范围,以加速模型训练的收敛速度,优化模型的学习效果。

划分数据集:将处理后的数据集划分为训练集、验证集和测试集,通常按照70%训练、20%验证、10%测试的比例进行分配,确保每个子集中的样本分布一致。

格式转换:将图像和标注数据转换为适合所用深度学习框架的格式,例如YOLO模型需要生成对应的TXT文件,记录每张图像中的手势信息。

标注格式:

VOC格式 (XML)

YOLO格式 (TXT)

yolo_dataset/

├── train/

│   ├── images/

│   │   ├── image1.jpg

│   │   ├── image2.jpg

│   │   ├── ...

│   │

│   └── labels/

│       ├── image1.txt

│       ├── image2.txt

│       ├── ...

└── test...

└── valid...

 

voc_dataset/

├── train/

│   ├───├

│   │   ├── image1.xml

│   │   ├── image2.xml

│   │   ├── ...

│   │

│   └───├

│       ├── image1.jpg

│       ├── image2.jpg

│       ├── ...

└── test...

└── valid...


三、模型训练

3.1理论技术

       卷积神经网络(CNN)是一种专门针对图像数据处理的深度学习模型,其结构主要由卷积层、激活层、池化层和全连接层组成。卷积层通过卷积操作提取输入图像的局部特征,

卷积核在图像上滑动,生成特征图。每个卷积核能够捕捉到不同类型的特征,如边缘、纹理和形状等,使得网络能够逐渐学习到更高层次的抽象特征。激活层(通常使用ReLU激活函数)

引入非线性,使得网络能够拟合更复杂的函数。池化层则通过下采样降低特征图的尺寸,减少计算量,从而提高模型的鲁棒性。最后,全连接层将提取的特征用于分类任务,输出最终的分类结果。

2.png

 CNN在手势数字识别中的主要优势在于其强大的特征提取能力和模型的自动学习能力。手势数字的形态各异,CNN能够通过层次化的特征学习自动提取出这些特征,从而实现准确分类。相比于传统图像处理方法,

CNN能够从大量标记数据中自动学习,减少了人工特征设计的需求。此外,CNN的参数共享机制使得模型在处理高维图像时更为高效,特别是在大规模数据集上进行训练时,能够迅速适应各种手势数字的变化,从而实现准确的实时检测。

3.png

 YOLO是一种基于回归的目标检测算法,其核心思想是将目标检测问题转化为回归问题。YOLO将输入图像划分为SxS的网格,每个网格负责预测物体的边界框及其类别概率。YOLO通过深度卷积神经网络,能够在单次前向传播中同时检测出

多个物体,显著提高了检测速度。该模型的网络结构包含多个卷积层与池化层,最终通过检测头输出每个网格的边界框坐标及置信度,同时预测物体的类别。这种方法的优点在于其全局上下文处理能力,能够理解图像中多个手势数字之间的关系。YOLO在手势数字识别中的优势体现在其实时性和高准确性。由于手势数字识别通常需要快速响应,YOLO能够以极高的速度处理图像并同时检测出不同的数字手势。

4.png

3.2模型训练

1. 数据集预处理


在实施YOLO项目之前,首先需要准备和划分数据集。数据集应包含不同手势数字的图像,以确保样本的多样性和代表性。可以通过拍摄或从网络下载获取样本。将数据集随机划分为训练集、验证集和测试集,通常推荐的比例为70%训练、20%验证、10%测试。以下是数据集划分的示例代码:


import os

import random

import shutil

 

# 定义数据集路径

dataset_path = 'path/to/hand_gesture_dataset'

images = os.listdir(dataset_path)

 

# 随机划分数据集

random.shuffle(images)

train_split = int(len(images) * 0.7)

val_split = int(len(images) * 0.9)

 

train_images = images[:train_split]

val_images = images[train_split:val_split]

test_images = images[val_split:]

 

# 创建新的目录以存放划分后的数据集

os.makedirs('train', exist_ok=True)

os.makedirs('val', exist_ok=True)

os.makedirs('test', exist_ok=True)

 

for image in train_images:

    shutil.copy(os.path.join(dataset_path, image), 'train/')

for image in val_images:

    shutil.copy(os.path.join(dataset_path, image), 'val/')

for image in test_images:

    shutil.copy(os.path.join(dataset_path, image), 'test/')


数据标注是YOLO项目中的关键环节,准确的标注直接影响模型的训练效果。使用LabelImg等标注工具为每张图像中的手势数字进行标注,通常采用矩形框的方式。标注过程的步骤如下:


启动LabelImg,选择需要标注的图像文件夹。

逐一打开图像,使用矩形工具绘制边界框,围绕不同的手势数字进行标注(如数字0到25)。

输入手势数字的类别名称(如“数字0”、“数字1”等)并保存标注。

确保每个手势数字均被准确标注,避免遗漏。

2. 模型训练


在完成数据准备后,需要配置YOLO模型。首先,准备模型的配置文件(如yolov3.cfg),设置网络参数、学习率和批量大小等。创建数据描述文件(如data.yaml),指定训练和验证数据集路径及类别数。例如,data.yaml文件内容如下:


train: train

val: val

nc: 26  # 目标类别数量(数字0到25)

names: ['数字0', '数字1', '数字2', '数字3', '数字4', '数字5', '数字6', '数字7', '数字8', '数字9', '数字10', '数字11', '数字12', '数字13', '数字14', '数字15', '数字16', '数字17', '数字18', '数字19', '数字20', '数字21', '数字22', '数字23', '数字24', '数字25']

模型配置完成后,可以开始训练YOLO模型。使用命令行运行YOLO训练命令,模型将开始处理训练数据。以下是训练的示例命令:


python train.py --img 640 --batch 16 --epochs 50 --data data.yaml --weights yolov3.weights

在训练过程中,可以根据需要调整学习率和其他超参数。通过命令行参数设置学习率:


python train.py --img 640 --batch 16 --epochs 50 --data data.yaml --weights yolov3.weights --hyp hyp.scratch.yaml

在hyp.scratch.yaml文件中,可以自定义学习率、动量、权重衰减等超参数:


# hyperparameters

lr0: 0.01  # 初始学习率

lrf: 0.1   # 最终学习率

momentum: 0.937  # 动量

weight_decay: 0.0005  # 权重衰减

3. 模型评估


完成训练后,对模型进行测试和评估是检验其性能的关键步骤。使用测试集中的图像,利用训练好的YOLO模型进行手势数字检测,生成检测结果并进行可视化。以下是测试和可视化的示例代码:


import cv2

import torch

 

# 加载训练好的模型

model = torch.hub.load('ultralytics/yolov5', 'custom', path='runs/train/exp/weights/best.pt')

 

# 进行检测

img = 'test/test_image.jpg'

results = model(img)

 

# 可视化检测结果

results.show()  # 显示结果

results.save('output/')  # 保存结果到指定目录

四、总结

       通过自制数据集和深度学习算法构建一个高效的手势数字识别检测系统。首先,通过精细的图像采集和标注,为模型训练提供丰富的样本支持。结合卷积神经网络和YOLO算法,构建了改进的CNN-YOLO模型,

以提升手势数字识别的准确性和速度。在各个阶段的实施中,通过合理的数据预处理和模型评估,验证了所构建模型的有效性,为手势数字识别提供了可靠的技术保障。

来源: 整理文章为传播相关技术,网络版权归原作者所有,如有侵权,请联系删除。



                           



                            



共1条 1/1 1 跳转至

回复

匿名不能发帖!请先 [ 登陆 注册 ]