TensorFlow实现:卷积神经网络识别手势动作(有代码与演示)

完整代码见我的GitHub地址:https://github.com/xingjianzhang1997/gesture-recognition

主要思路:1.构建数据集          2.建立神经网络         3.训练并调参        4.保存并调用

另外我还加了界面设计和机械臂控制(通过串口通信把判断结果发给单片机,所以代码中有serial模块)

效果演示:

                              

              

project清单:

一共识别5种手势动作 1. 剪刀动作 2.石头动作 3.布动作 4.OK动作 5.good动作

训练集: 1~4号动作各有1300张照片,5号动作有1450张照片 测试集: 1~5号动作各有200张照片

文件解释:

1.data文件里面存放test数据集,train数据集,实时保存的图像(用于在线检测)。

2.ges_ico文件里面存放界面窗口显示的各种图标。

3.log文件里面存放训练好的CNN网络的模型参数。

4.CallFrame.py是界面窗口的逻辑文件,用来调用界面文件并编写信号与槽函数。

5.Frame.py是界面窗口的界面文件,通过PyQt5的designer工具生成。

6.GetTestImage.py是利用OpenCV获取图片并标记,用来制作测试集。

7.GetTrainImage.py是利用OpenCV获取图片并标记,用来制作训练集。

8.SaveGesture.py是利用OpenCV实时获取图片,并进行预处理,用于在线检测手势。

9.TestGesture.py是将实时获取的图片送入已训练好的CNN中判断其手势动作。

10.TestInTest.py是将测试集送入已训练好的CNN中判断该网络模型的准确率。

11.Train.py是训练CNN模型函数,并将训练好的模型参数保存在本地。

12.Train_inputdata.py是用来读取数据集的图像和标签,并打包成batch形式。

13.Train_model.py是模型结构,这里用的是AlexNet结构。

使用方法: 先用Train.py训练好模型参数,然后运行CallFrame.py调用出界面窗口, 点击窗口的相应按钮就可以在线检测手势动作,其中的执行手势按钮是和下位机通信(如STM32),通过串口函数将识别结果传给下位机,实现根据手势动作控制的功能。

测试结果: 使用该模型训练到900步的时候在测试集上正确率可以稳定在95%左右。

  • 44
    点赞
  • 379
    收藏
    觉得还不错? 一键收藏
  • 143
    评论
以下是一个使用TensorFlow实现神经网络的示例代码: ``` python import tensorflow as tf # 定义输入和输出 input = tf.placeholder(dtype=tf.float32, shape=[None, 28, 28, 1]) output = tf.placeholder(dtype=tf.float32, shape=[None, 10]) # 定义卷层 conv1 = tf.layers.conv2d(inputs=input, filters=32, kernel_size=[5, 5], padding="same", activation=tf.nn.relu) # 定义池化层 pool1 = tf.layers.max_pooling2d(inputs=conv1, pool_size=[2, 2], strides=2) # 定义第二个卷和池化层 conv2 = tf.layers.conv2d(inputs=pool1, filters=64, kernel_size=[5, 5], padding="same", activation=tf.nn.relu) pool2 = tf.layers.max_pooling2d(inputs=conv2, pool_size=[2, 2], strides=2) # 将卷和池化层输出扁平化 flatten = tf.layers.flatten(inputs=pool2) # 定义全连接层 fc1 = tf.layers.dense(inputs=flatten, units=1024, activation=tf.nn.relu) # 添加dropout层,以防止过拟合 dropout = tf.layers.dropout(inputs=fc1, rate=0.4) # 定义输出层 logits = tf.layers.dense(inputs=dropout, units=10) # 定义损失函数和优化器 loss = tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits_v2(logits=logits, labels=output)) optimizer = tf.train.AdamOptimizer().minimize(loss) # 训练模型 with tf.Session() as sess: sess.run(tf.global_variables_initializer()) for i in range(num_epochs): # 在这里添加代码以将数据馈送到模型中,执行训练操作,并计算损失 ``` 请注意,这只是一个示例代码,并且需要根据您的具体问题和数据进行修改。
评论 143
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值