Thenujan commited on
Commit
06758a2
·
1 Parent(s): 884b2ca

added model

Browse files
Files changed (2) hide show
  1. .gitignore +1 -0
  2. similarity_search.py +4 -2
.gitignore CHANGED
@@ -0,0 +1 @@
 
 
1
+ model1.pt
similarity_search.py CHANGED
@@ -89,11 +89,13 @@ def get_feature_vector_img(model, imgs, epoch=10, use_cuda=False):
89
  return np.concatenate(features, axis=0)
90
 
91
  def Model():
92
- backbone = open_clip.create_model_and_transforms('ViT-H-14', None)[0].visual
93
- backbone.load_state_dict(th.load('./model1.pt')) #https://huggingface.co/hca97/aicrowd-visual-recognition-models/blob/main/model1.pt
 
94
  backbone.eval() # Dropping unecessary layers
95
  return backbone
96
 
 
97
  def find(image):
98
  # img_path = '/kaggle/input/vprtestdata/public_dataset/queries/abiding-inchworm-of-ultimate-freedom.jpeg'
99
 
 
89
  return np.concatenate(features, axis=0)
90
 
91
  def Model():
92
+ # backbone = open_clip.create_model_and_transforms('ViT-H-14', None)[0].visual
93
+ # backbone.load_state_dict(th.load('./model1.pt')) #https://huggingface.co/hca97/aicrowd-visual-recognition-models/blob/main/model1.pt
94
+ backbone, preprocess_train, preprocess_val = open_clip.create_model_and_transforms('hf-hub:Thenujan/ViT-H-14')
95
  backbone.eval() # Dropping unecessary layers
96
  return backbone
97
 
98
+
99
  def find(image):
100
  # img_path = '/kaggle/input/vprtestdata/public_dataset/queries/abiding-inchworm-of-ultimate-freedom.jpeg'
101