update readme to refer to new pretrained models

google · Apr 5, 2021 · c58daa0 · c58daa0
1 parent c27ca39
commit c58daa0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/README.md b/README.md
@@ -138,6 +138,8 @@ Our pretrained models can be downloaded below. You can find our best single mode
 - [MITStates Model](https://storage.googleapis.com/image_retrieval_css/pretrained_models/checkpoint_mitstates.pth): 0.132
 
 
+These saved weights might not be working correctly any more with new version, please refer to https://github.com/google/tirg/issues/12
+
 
 ## Notes:
 All log files will be saved at `./runs/<timestamp><comment>`.

diff --git a/test_retrieval.py b/test_retrieval.py
@@ -97,12 +97,6 @@ def test(opt, model, testset):
     all_imgs[i, :] /= np.linalg.norm(all_imgs[i, :])
 
   # match test queries to target images, get nearest neighbors
-  # sims = all_queries.dot(all_imgs.T)
-  #sims = []
-  #if test_queries:
-  #  for i, t in enumerate(test_queries):
-  #    sims[i, t['source_img_id']] = -10e10  # remove query image
-  #nn_result = [np.argsort(-sims[i, :])[:110] for i in range(sims.shape[0])]
   nn_result = []
   for i in tqdm(range(all_queries.shape[0])):
     sims = all_queries[i:(i+1), :].dot(all_imgs.T)