Skip to content

Commit a20b3f8

Browse files
author
Anastasia Murzova
committed
Added data for TF nearest neighbour resize layer test
1 parent 665e00b commit a20b3f8

5 files changed

+10
-0
lines changed

testdata/dnn/tensorflow/generate_tf_models.py

+10
Original file line numberDiff line numberDiff line change
@@ -318,6 +318,16 @@ def my_dropout(x):
318318
scaled = tf.image.resize_nearest_neighbor(scaled, size=(9, 12))
319319
save(inp, scaled, 'resize_nearest_neighbor')
320320
################################################################################
321+
inp = tf.placeholder(tf.float32, [2, 10, 11, 3], 'input')
322+
resized = tf.image.resize_nearest_neighbor(inp, size=(15, 13), align_corners=True, name='resize_nearest_neighbor',
323+
half_pixel_centers=False)
324+
save(inp, resized, ('resize_nearest_neighbor', 'align_corners'), is_gen_data=False)
325+
################################################################################
326+
inp = tf.placeholder(tf.float32, [2, 10, 11, 3], 'input')
327+
resized = tf.image.resize_nearest_neighbor(inp, size=(15, 13), align_corners=False, name='resize_nearest_neighbor',
328+
half_pixel_centers=True)
329+
save(inp, resized, ('resize_nearest_neighbor', 'half_pixel'), is_gen_data=False)
330+
################################################################################
321331
inp = tf.placeholder(tf.float32, [1, 2, 3, 4], 'input')
322332
bn = tf.layers.batch_normalization(inp, training=isTraining, fused=False, name='batch_norm',
323333
beta_initializer=tf.random_normal_initializer(),
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.

0 commit comments

Comments
 (0)