demo scripts - bluekingsong/vision_material GitHub Wiki

#separable_convolution2d

import numpy as np
import cv2
import tensorflow as tf
from tensorflow.contrib import slim

deepth_0 = tf.constant(np.ones([2, 2]))
deepth_1 = tf.constant(2 * np.ones([2, 2]))

inputs = tf.expand_dims(tf.stack([deepth_0, deepth_1], -1), 0)
print inputs.shape

outputs = slim.separable_convolution2d(inputs, None, 1, 2, activation_fn=None,
weights_initializer=tf.ones_initializer)

print ‘outputs.shape=’, outputs.shape

with tf.Session() as sess:
sess.run(tf.global_variables_initializer())
print sess.run(outputs[:, :, :, 0:2])
print sess.run(outputs[:, :, :, 2:4])