diff --git a/training/training_loop.py b/training/training_loop.py index 62716b36133b1ed868dcf9511577d3f2e8f66fe7..61264e1bac1158b9d81ca99aef7c16aa6d220126 100755 --- a/training/training_loop.py +++ b/training/training_loop.py @@ -73,16 +73,11 @@ def process_reals(x, labels, lod, mirror_augment, mirror_augment_v, spatial_augm x = tf.where(tf.random_uniform([tf.shape(x)[0]]) < 0.5, x, tf.reverse(x, [2])) if spatial_augmentations: with tf.name_scope('SpatialAugmentations'): - print('DATA SHAPE:') - print(dshape) x.set_shape(dshape) x = tf.transpose(x, [0, 2, 3, 1]) - print('BEFORE: ',x) x = tf.map_fn(apply_random_aug, x) - print('AFTER: ',x) x = tf.transpose(x, [0, 3, 1, 2]) - print('POST AUGMENT/TRANSPOSE :') - print(x.get_shape()) + with tf.name_scope('FadeLOD'): # Smooth crossfade between consecutive levels-of-detail. s = tf.shape(x) y = tf.reshape(x, [-1, s[1], s[2]//2, 2, s[3]//2, 2])