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 splitlines使用技巧
Sep 06 Python
python list中append()与extend()用法分享
Mar 24 Python
python文件写入实例分析
Apr 08 Python
Python中数组,列表:冒号的灵活用法介绍(np数组,列表倒序)
Apr 18 Python
Python中如何使用if语句处理列表实例代码
Feb 24 Python
Python私有属性私有方法应用实例解析
Sep 15 Python
python实现人工蜂群算法
Sep 18 Python
python pip如何手动安装二进制包
Sep 30 Python
用python写PDF转换器的实现
Oct 29 Python
python 使用tkinter+you-get实现视频下载器
Nov 17 Python
Numpy中np.max的用法及np.maximum区别
Nov 27 Python
分享PyCharm最新激活码(真永久激活方法)不用每月找安装参数或最新激活码了
Dec 27 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
Windows下PHP的任意文件执行漏洞
2006/10/09 PHP
php学习之数据类型之间的转换介绍
2011/06/09 PHP
分享下页面关键字抓取components.arrow.com站点代码
2014/01/30 PHP
微信支付扫码支付php版
2016/07/22 PHP
PHP绕过open_basedir限制操作文件的方法
2018/06/10 PHP
php正确输出json数据的实例讲解
2018/08/21 PHP
PHP数字金额转换成中文大写显示
2019/01/05 PHP
JavaScript 调试器简介
2009/02/21 Javascript
超越Jquery_01_isPlainObject分析与重构
2010/10/20 Javascript
jQuery中json对象的复制方式介绍(数组及对象)
2013/06/08 Javascript
Web前端新人笔记之jquery入门心得(新手必看)
2016/05/17 Javascript
js print打印网页指定区域内容的简单实例
2016/11/01 Javascript
解决JS内存泄露之js对象和dom对象互相引用问题
2017/06/25 Javascript
jQuery回调方法使用示例
2017/06/26 jQuery
解决layui中的form表单与button的点击事件冲突问题
2018/08/15 Javascript
bootstrap+spring boot实现面包屑导航功能(前端代码)
2019/10/09 Javascript
微信小程序自定义tabbar custom-tab-bar 6s出不来解决方案(cover-view不兼容)
2019/11/01 Javascript
小程序分享链接onShareAppMessage的具体用法
2020/05/22 Javascript
Javascript实现鼠标移入方向感知
2020/06/24 Javascript
[56:18]DOTA2上海特级锦标赛主赛事日 - 4 败者组第四轮#2 MVP.Phx VS Fnatic第二局
2016/03/05 DOTA
详解Python中的Descriptor描述符类
2016/06/14 Python
简单了解Django模板的使用
2017/12/20 Python
python基于SMTP协议发送邮件
2019/05/31 Python
Python学习笔记之抓取某只基金历史净值数据实战案例
2019/06/03 Python
python try except返回异常的信息字符串代码实例
2019/08/15 Python
Django CSRF认证的几种解决方案
2020/03/03 Python
CSS3中border-radius属性设定圆角的使用技巧
2016/05/10 HTML / CSS
中学运动会广播稿
2014/01/19 职场文书
学校做一个有道德的人活动方案
2014/08/23 职场文书
自主招生学校推荐信
2014/09/26 职场文书
师范生见习报告
2014/10/31 职场文书
2015年班组建设工作总结
2015/05/13 职场文书
雷锋的观后感
2015/06/10 职场文书
2016年清明节红领巾广播稿
2015/12/17 职场文书
导游词之杭州岳王庙
2019/11/13 职场文书
TV动画《神废柴☆偶像》公布先导PV
2022/03/20 日漫