From 973d49fdde187462d92d56cb3a01b94872a88b68 Mon Sep 17 00:00:00 2001
From: sdtblck <sdtblck@gmail.com>
Date: Tue, 16 Jun 2020 23:09:49 +0100
Subject: [PATCH] -
---
training/training_loop.py | 7 +------
1 file changed, 1 insertion(+), 6 deletions(-)
diff --git a/training/training_loop.py b/training/training_loop.py
index 62716b3..61264e1 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])
--
GitLab