From 179f98c949dbeeb383d152e46610dd6789616231 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Sat, 11 Jan 2020 00:23:46 +0100 Subject: upload video script --- cli/app/search/search_dense.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) (limited to 'cli/app/search/search_dense.py') diff --git a/cli/app/search/search_dense.py b/cli/app/search/search_dense.py index 3719de6..ac67d07 100644 --- a/cli/app/search/search_dense.py +++ b/cli/app/search/search_dense.py @@ -240,18 +240,18 @@ def find_dense_embedding_for_images(params): feat_loss += tf.reduce_mean(feat_square_diff) * 0.25 img_feat_err += tf.reduce_mean(feat_square_diff, axis=1) * 0.25 - # gen_feat = gen_feat_ex["InceptionV3/Conv2d_3b_1x1"] - # target_feat = target_feat_ex["InceptionV3/Conv2d_3b_1x1"] - # feat_square_diff = tf.reshape(tf.square(gen_feat - target_feat), [BATCH_SIZE, -1]) - # feat_loss += tf.reduce_mean(feat_square_diff) * 0.25 - # img_feat_err += tf.reduce_mean(feat_square_diff, axis=1) * 0.25 - - gen_feat = gen_feat_ex["InceptionV3/Mixed_6a"] - target_feat = target_feat_ex["InceptionV3/Mixed_6a"] + gen_feat = gen_feat_ex["InceptionV3/Conv2d_3b_1x1"] + target_feat = target_feat_ex["InceptionV3/Conv2d_3b_1x1"] feat_square_diff = tf.reshape(tf.square(gen_feat - target_feat), [BATCH_SIZE, -1]) feat_loss += tf.reduce_mean(feat_square_diff) * 0.25 img_feat_err += tf.reduce_mean(feat_square_diff, axis=1) * 0.25 + # gen_feat = gen_feat_ex["InceptionV3/Mixed_6a"] + # target_feat = target_feat_ex["InceptionV3/Mixed_6a"] + # feat_square_diff = tf.reshape(tf.square(gen_feat - target_feat), [BATCH_SIZE, -1]) + # feat_loss += tf.reduce_mean(feat_square_diff) * 0.25 + # img_feat_err += tf.reduce_mean(feat_square_diff, axis=1) * 0.25 + gen_feat = gen_feat_ex["InceptionV3/Mixed_7a"] target_feat = target_feat_ex["InceptionV3/Mixed_7a"] feat_square_diff = tf.reshape(tf.square(gen_feat - target_feat), [BATCH_SIZE, -1]) -- cgit v1.2.3-70-g09d2