Spaces:
Sleeping
Sleeping
added model
Browse files- .gitignore +1 -0
- 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 |
|