summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--cli/app/search/search_dense.py10
1 files changed, 5 insertions, 5 deletions
diff --git a/cli/app/search/search_dense.py b/cli/app/search/search_dense.py
index fb74411..10dbbe0 100644
--- a/cli/app/search/search_dense.py
+++ b/cli/app/search/search_dense.py
@@ -584,11 +584,11 @@ def feature_loss_vgg(feature_extractor, opt_feature_layers, BATCH_SIZE, img_a, i
img_b = tf.image.resize_images(img_b, [resize_height, resize_width])
global scope_index
- # scope_index += 1
- # scope_a = 'vgg_16_{}_a'.format(scope_index)
- # scope_b = 'vgg_16_{}_b'.format(scope_index)
- scope_a = 'vgg_16'
- scope_b = 'vgg_16'
+ scope_index += 1
+ scope_a = 'vgg_16_{}_a'.format(scope_index)
+ scope_b = 'vgg_16_{}_b'.format(scope_index)
+ # scope_a = 'vgg_16'
+ # scope_b = 'vgg_16'
gen_fc, gen_feat_ex = nets.vgg.vgg_16(img_a, scope=scope_a) #, reuse=True)
target_fc, target_feat_ex = nets.vgg.vgg_16(img_b, scope=scope_b) #, reuse=True)
# gen_feat_ex = feature_extractor(dict(images=img_a), as_dict=True, signature='image_feature_vector')