diff options
Diffstat (limited to 'cli/app/search/search_dense.py')
| -rw-r--r-- | cli/app/search/search_dense.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/cli/app/search/search_dense.py b/cli/app/search/search_dense.py index 18c430e..e5769ab 100644 --- a/cli/app/search/search_dense.py +++ b/cli/app/search/search_dense.py @@ -588,8 +588,8 @@ def feature_loss_vgg(feature_extractor, opt_feature_layers, BATCH_SIZE, img_a, i 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_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_fc, gen_feat_ex = vgg_model(img_a) #, reuse=True) |