tensorflow 实现自定义layer并添加到计算图中


Posted in Python onFebruary 04, 2020

目的

将用户自定义的layer结合tensorflow自带的layer组成多层layer的计算图。

实现功能

对2D图像进行滑动窗口平均,并通过自定义的操作layer返回结果。

import tensorflow as tf
import numpy as np
sess = tf.Session()

#将size设为[1, 4, 4, 1]是因为tf中图像函数是处理四维图片的。
#这四维依次是: 图片数量,高度, 宽度, 颜色通道
x_shape = [1,4,4,1]
x_val = np.random.uniform(size = x_shape)


#tf.nn.conv2d中name表明该layer命名为“Moving_Avg_Window”
#该卷积核为[[0.25,0.25],[0.25,0.25]],所以是一个求平均操作
x_data = tf.placeholder(tf.float32, shape = x_shape)
my_filter = tf.constant(0.25, shape = [2,2,1,1])
my_strides = [1,2,2,1]
mov_avg_layer = tf.nn.conv2d(x_data, my_filter, my_strides, padding = 'SAME', name = 'Moving_Avg_Window')


#自定义layer,对卷积操作之后的输出做操作
def custom_layer(input_matrix):
  input_matrix_sqeeze = tf.squeeze(input_matrix)
  A = tf.constant([1.,2.],[-1.,3.])
  b = tf.constant(1., shape = [2,2])
  temp1 = tf.matmul(A, input_matrix_sqeeze)
  temp2 = tf.add(temp1, b)
  return(tf.sigmod(temp2))
#把刚刚自定义的layer加入到计算图中,并给予自定义的命名(利用tf.name_scope())
with tf.name_scope('Custom_Layer') as scope:
  custom_layer1 = custom_layer(mov_avg_layer)


#为占位符传入4*4图片,并执行计算图
print(sess.run(custom_layer, feed_dict= {x_data: x_val}))

以上这篇tensorflow 实现自定义layer并添加到计算图中就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
在Django中进行用户注册和邮箱验证的方法
May 09 Python
Python爬取网易云音乐热门评论
Mar 31 Python
django 多数据库配置教程
May 30 Python
朴素贝叶斯分类算法原理与Python实现与使用方法案例
Jun 26 Python
Python中正反斜杠(‘/’和‘\’)的意义与用法
Aug 12 Python
python中property和setter装饰器用法
Dec 19 Python
tensorflow实现打印ckpt模型保存下的变量名称及变量值
Jan 04 Python
基于python实现语音录入识别代码实例
Jan 17 Python
浅谈Tensorflow 动态双向RNN的输出问题
Jan 20 Python
在python tkinter界面中添加按钮的实例
Mar 04 Python
Python 判断时间是否在时间区间内的实例
May 16 Python
python使用QQ邮箱实现自动发送邮件
Jun 22 Python
TensorFlow实现自定义Op方式
Feb 04 #Python
tensorflow使用指定gpu的方法
Feb 04 #Python
TensorFlow梯度求解tf.gradients实例
Feb 04 #Python
基于TensorFlow中自定义梯度的2种方式
Feb 04 #Python
tensorflow 查看梯度方式
Feb 04 #Python
opencv python图像梯度实例详解
Feb 04 #Python
TensorFlow设置日志级别的几种方式小结
Feb 04 #Python
You might like
解析PHP提交后跳转
2013/06/23 PHP
PHP中使用循环实现的金字塔图形
2014/11/08 PHP
Redis使用Eval多个键值自增的操作实例
2016/11/04 PHP
在textarea中显示html页面的javascript代码
2007/04/20 Javascript
JQuery团队打造的javascript单元测试工具QUnit介绍
2010/02/26 Javascript
jQuery 顺便学习下CSS选择器 奇偶匹配nth-child(even)
2010/05/24 Javascript
jquery ajax例子返回值详解
2012/09/11 Javascript
jQuery Ajax使用实例
2015/04/16 Javascript
JS封装的自动创建表格的实现代码
2016/06/15 Javascript
BootStrap selectpicker
2016/06/20 Javascript
javascript弹出窗口中增加确定取消按钮
2016/06/24 Javascript
利用jQuery对无序列表排序的简单方法
2016/10/16 Javascript
基于slideout.js实现移动端侧边栏滑动特效
2016/11/28 Javascript
JavaScript页面倒计时功能完整示例
2019/05/15 Javascript
如何在 ant 的table中实现图片的渲染操作
2020/10/28 Javascript
[02:51]2018年度DOTA2最佳中单位选手-完美盛典
2018/12/17 DOTA
Python中的__new__与__init__魔术方法理解笔记
2014/11/08 Python
Python实现计算两个时间之间相差天数的方法
2017/05/10 Python
使用python验证代理ip是否可用的实现方法
2018/07/25 Python
Pyqt5如何让QMessageBox按钮显示中文示例代码
2019/04/11 Python
python异步编程 使用yield from过程解析
2019/09/25 Python
python实现XML解析的方法解析
2019/11/16 Python
python做接口测试的必要性
2019/11/20 Python
Pyecharts绘制全球流向图的示例代码
2020/01/08 Python
Python3中的f-Strings增强版字符串格式化方法
2020/03/04 Python
Python 去除字符串中指定字符串
2020/03/05 Python
pyspark 随机森林的实现
2020/04/24 Python
python time()的实例用法
2020/11/03 Python
Django数据模型中on_delete使用详解
2020/11/30 Python
使用phonegap进行本地存储的实现方法
2017/03/31 HTML / CSS
沪江旗下的海量优质课程平台:沪江网校
2017/11/07 全球购物
应届大学生简历中的自我评价
2014/01/15 职场文书
公务员诚信承诺书
2014/05/26 职场文书
住房租房协议书
2014/08/20 职场文书
pytorch--之halfTensor的使用详解
2021/05/24 Python
MySQL表字段数量限制及行大小限制详情
2022/07/23 MySQL