diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2020-02-16 10:58:08 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2020-02-16 10:58:08 +0100 |
| commit | 920767526bc39277b6069368f42d57e137ed93a8 (patch) | |
| tree | 76ecaa2e70d9a68c1bee89e12a1be0a46970b1b6 /cli/app/search | |
| parent | 3835bd1700a3f36d10d66ac9358b4aa4fbfeaa10 (diff) | |
scope
Diffstat (limited to 'cli/app/search')
| -rw-r--r-- | cli/app/search/search_dense.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/cli/app/search/search_dense.py b/cli/app/search/search_dense.py index 797f8d5..bb6f636 100644 --- a/cli/app/search/search_dense.py +++ b/cli/app/search/search_dense.py @@ -567,7 +567,7 @@ def feature_loss_tfhub(feature_extractor, opt_feature_layers, BATCH_SIZE, img_a, return feat_loss / len(opt_feature_layers), img_feat_err / len(opt_feature_layers) scope_index = 0 -vgg_model = tf.make_template(vgg.vgg_16) +vgg_model = tf.make_template(nets.vgg.vgg_16) def feature_loss_vgg(feature_extractor, opt_feature_layers, BATCH_SIZE, img_a, img_b, y, x, height, width, resize_height=None, resize_width=None): height = int(height) width = int(width) |