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 相关文章推荐
Python基于numpy灵活定义神经网络结构的方法
Aug 19 Python
Python使用当前时间、随机数产生一个唯一数字的方法
Sep 18 Python
Python编程使用*解包和itertools.product()求笛卡尔积的方法
Dec 18 Python
python如何重载模块实例解析
Jan 25 Python
使用requests库制作Python爬虫
Mar 25 Python
Python smtplib实现发送邮件功能
May 22 Python
使用pyshp包进行shapefile文件修改的例子
Dec 06 Python
通过python调用adb命令对App进行性能测试方式
Apr 23 Python
Python Django搭建网站流程图解
Jun 13 Python
利用Selenium添加cookie实现自动登录的示例代码(fofa)
May 08 Python
写好Python代码的几条重要技巧
May 21 Python
python爬取豆瓣电影TOP250数据
May 23 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
Protoss兵种介绍
2020/03/14 星际争霸
php析构函数的简单使用说明
2015/08/24 PHP
CI框架(CodeIgniter)公共模型类定义与用法示例
2017/08/10 PHP
laravel框架中表单请求类型和CSRF防护实例分析
2019/11/23 PHP
JavaScript为对象原型prototype添加属性的两种方式
2010/08/01 Javascript
巧用jquery解决下拉菜单被Div遮挡的相关问题
2014/02/13 Javascript
微信支付如何实现内置浏览器的H5页面支付
2015/09/25 Javascript
javascript设计模式--策略模式之输入验证
2015/11/27 Javascript
jQuery ajax提交Form表单实例(附demo源码)
2016/04/06 Javascript
Bootstrap教程JS插件弹出框学习笔记分享
2016/05/17 Javascript
浅析BootStrap Treeview的简单使用
2016/10/12 Javascript
微信小程序一周时间表功能实现
2019/10/17 Javascript
js+css实现全屏侧边栏
2020/06/16 Javascript
vue 遮罩层阻止默认滚动事件操作
2020/07/28 Javascript
uniapp微信小程序:key失效的解决方法
2021/01/20 Javascript
[01:04]不如跳舞!DOTA2新英雄玛尔斯的欢乐日常
2019/03/11 DOTA
详解Python Socket网络编程
2016/01/05 Python
简单谈谈Python中的几种常见的数据类型
2017/02/10 Python
Django 导出 Excel 代码的实例详解
2017/08/11 Python
Python语言描述连续子数组的最大和
2018/01/04 Python
浅谈使用Python内置函数getattr实现分发模式
2018/01/22 Python
用pandas按列合并两个文件的实例
2018/04/12 Python
python turtle库画一个方格和圆实例
2019/06/27 Python
Python的PIL库中getpixel方法的使用
2020/04/09 Python
Pycharm中安装wordcloud等库失败问题及终端通过pip安装的Python库如何添加到Pycharm解释器中(推荐)
2020/05/10 Python
使用keras实现BiLSTM+CNN+CRF文字标记NER
2020/06/29 Python
html5服务器推送_动力节点Java学院整理
2017/07/12 HTML / CSS
亚马逊西班牙购物网站:amazon西班牙
2017/03/06 全球购物
意大利折扣和优惠券网站:Groupalia
2019/10/09 全球购物
网络工程专业毕业生推荐信
2013/10/28 职场文书
办公室主任个人对照检查材料思想汇报
2014/10/11 职场文书
文体活动总结
2015/02/04 职场文书
2015年母亲节寄语
2015/03/23 职场文书
会议开幕致辞怎么写
2016/03/03 职场文书
字典算法实现及操作 --python(实用)
2021/03/31 Python
python 实现mysql自动增删分区的方法
2021/04/01 Python