add bit model

src/similarity/model_implements/bit.py

@@ -0,0 +1,13 @@
+import tensorflow_hub as hub
+import numpy as np 
+
+class BigTransfer:
+
+    def __init__(self):
+        self.module = hub.KerasLayer("https://tfhub.dev/google/bit/m-r50x1/1")
+
+    def extract_feature(self, imgs):
+        features = []
+        for img in imgs:
+            features.append(np.squeeze(self.module(img)))
+        return features

src/similarity/similarity.py

@@ -3,17 +3,18 @@ from src.util import image as image_util
 from src.util import matrix
 from .model_implements.mobilenet_v3 import ModelnetV3
 from .model_implements.vit_base import VitBase
+from .model_implements.bit import BigTransfer
 
 
 class Similarity:
     def get_models(self):
         return [
             model.SimilarityModel(name= 'Mobilenet V3', image_size= 224, model_cls = ModelnetV3()),
+            model.SimilarityModel(name= 'Big Transfer (BiT)', image_size= 224, model_cls = BigTransfer()),
             model.SimilarityModel(name= 'Vision Transformer', image_size= 224, model_cls = VitBase(), image_input_type='pil'),
             ]        
 
     def check_similarity(self, img_urls, model):
-        # model = self.get_models()[model_idx]
         imgs = []
         for url in img_urls:
             if url == "": continue