如下卷积核具有边缘检测的作用,举例说明:
[[-1,-1,-1],
[-1,8,-1],
[-1,-1,-1]]

import cv2
import numpy as np
import tensorflow as tf

input_data = cv2.imread('zxc.jpg')
weights_data = [[[-1,-1,-1],[-1,-1,-1],[-1,-1,-1]],
		[[-1,-1,-1],[8,8,8],[-1,-1,-1]],
		[[-1,-1,-1],[-1,-1,-1],[-1,-1,-1]]]

def tf_conv2d(input,weights):
	conv = tf.nn.conv2d(input, weights, strides=[1, 1, 1, 1], padding='SAME')
	return conv

def main(): 
	const_input = tf.constant(input_data , tf.float32)
	input = tf.Variable(const_input,name="input")
	# [1500,1000,3]-->[1,1500,1000,3]
	input=tf.expand_dims(input,0)

	const_weights = tf.constant(weights_data , tf.float32 )
	weights = tf.Variable(const_weights,name="weights")
	#[3,3,3]-->[3,3,3,1]
	weights=tf.expand_dims(weights,3) 

	#[b,h,w,c]
	conv=tf_conv2d(input,weights)

	init=tf.global_variables_initializer()

	with tf.Session() as sess:
		sess.run(init)
		imgs = sess.run(conv)
		cv2.imwrite('zxcl.jpg', imgs[0])

if __name__ == '__main__':
	main()

zxc.jpg如下:
在这里插入图片描述
zxcl.jpg如下:
在这里插入图片描述

GitHub 加速计划 / te / tensorflow
184.55 K
74.12 K
下载
一个面向所有人的开源机器学习框架
最近提交(Master分支:2 个月前 )
a49e66f2 PiperOrigin-RevId: 663726708 2 个月前
91dac11a This test overrides disabled_backends, dropping the default value in the process. PiperOrigin-RevId: 663711155 2 个月前
Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐