diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2019-12-14 22:57:44 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2019-12-14 22:57:44 +0100 |
| commit | 3ca730b3f0d109ffb8673d5f266159e752fa44a4 (patch) | |
| tree | 1389d0e4cae9a2af7ac94fa980d673a612680748 | |
| parent | 28bc61bd7aa9f6bb2de1dccb96a4074b89eef946 (diff) | |
nutha experiment
| -rw-r--r-- | inversion/image_inversion_inception.py | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/inversion/image_inversion_inception.py b/inversion/image_inversion_inception.py index 1b7b349..bb883e7 100644 --- a/inversion/image_inversion_inception.py +++ b/inversion/image_inversion_inception.py @@ -226,11 +226,11 @@ if params.features: feat_loss = tf.reduce_mean(feat_square_diff) * 0.5 img_feat_err = tf.reduce_mean(feat_square_diff, axis=1) * 0.5 - gen_feat = gen_feat_ex["InceptionV3/Mixed_6a"] - target_feat = target_feat_ex["InceptionV3/Mixed_6a"] + gen_feat = gen_feat_ex["InceptionV3/Mixed_6b"] + target_feat = target_feat_ex["InceptionV3/Mixed_6b"] feat_square_diff = tf.reshape(tf.square(gen_feat - target_feat), [BATCH_SIZE, -1]) - feat_loss += tf.reduce_mean(feat_square_diff) * 0.34 - img_feat_err += tf.reduce_mean(feat_square_diff, axis=1) * 0.25 + feat_loss += tf.reduce_mean(feat_square_diff) * 0.3 + img_feat_err += tf.reduce_mean(feat_square_diff, axis=1) * 0.3 # gen_feat = gen_feat_ex["InceptionV3/Mixed_5a"] # target_feat = target_feat_ex["InceptionV3/Mixed_5a"] @@ -247,8 +247,8 @@ if params.features: gen_feat = gen_feat_ex["InceptionV3/Mixed_7c"] target_feat = target_feat_ex["InceptionV3/Mixed_7c"] feat_square_diff = tf.reshape(tf.square(gen_feat - target_feat), [BATCH_SIZE, -1]) - feat_loss += tf.reduce_mean(feat_square_diff) * 0.34 - img_feat_err += tf.reduce_mean(feat_square_diff, axis=1) * 0.25 + feat_loss += tf.reduce_mean(feat_square_diff) * 0.2 + img_feat_err += tf.reduce_mean(feat_square_diff, axis=1) * 0.2 else: feat_loss = tf.constant(0.0) |