Tensorflow 实现卷积操作

最新推荐文章于 2023-08-03 11:46:00 发布

GuoSmallGuo23

最新推荐文章于 2023-08-03 11:46:00 发布

阅读量1.1k

点赞数 1

分类专栏： Deep Learning 文章标签： Tensorflow 卷积深度学习手部关键点检测

本文链接：https://blog.csdn.net/yonglingqin9031/article/details/86761259

版权

本文通过作者在手部关键点检测项目中遇到的Tensorflow卷积操作，详细解释了如何实现卷积和ReLU激活函数。代码分为ReLU实现、卷积实现和卷积+ReLU三部分，包括输入tensor的尺寸理解、卷积核初始化、步长设置以及核心的卷积函数tf.nn.conv2d()的使用。文章还简单介绍了添加偏置的操作，并提供了测试流程。

摘要由CSDN通过智能技术生成

Tensorflow 实现卷积操作

恰逢新年假期，家里的事情变得多了起来。但是十几天零星的时间如果拿来积累，总是有些用的。近期着手于手部关键点检测，读了一篇论文 ICCV2017 Learning to Estimate 3D Hand Pose from Single RGB Images 并且查看了源代码。之前在实验室里自己录制了视频进行了简单的测试，但是结果差强人意。说跑题了，作者的代码功底应该是很深厚的，下面我引用一下作者书写的代码，来简单说明一下Tensorflow是如何实现卷积操作的。

这里我罗列一下作者的代码：

import tensorflow as tf

neg_slope_of_relu = 0.01

//Relu operation
def leaky_relu(tensor, name='relu'):
	out_tensor = tf.maximum(tensor, neg_slope_of_relu * tensor, name=name)
	return out_tensor

//Conv operation
def conv(in_tensor, layer_name, kernel_size, stride, out_chan, trainable=True):
	with tf.variable_scope(layer_name):
		in_size = in_tensor.get_shape