add more models

app.py

@@ -14,6 +14,7 @@ import logging
 from mmengine.logging.logger import MMFormatter
 from mmcls import list_models as list_models_
 from mmcls.apis.model import ModelHub, init_model
+import os
 
 
 @st.cache()
@@ -39,6 +40,7 @@ def list_models(*args, **kwargs):
 DATA_ROOT = '.'
 ANNO_FILE = 'meta/val.txt'
 LOG_FILE = 'demo.log'
+CACHED_PATH = 'cache'
 
 
 def get_model(model_name, pretrained=True):
@@ -65,7 +67,7 @@ def get_model(model_name, pretrained=True):
     cfg.model = new_model_cfg
 
     # prepare prototype
-    cached_path = f'cache/{model_name}_prototype.pt'  # noqa
+    cached_path = f'{CACHED_PATH}/{model_name}_prototype.pt'  # noqa
     cfg.model.prototype = cached_path
 
     model = init_model(metainfo.config, None, device='cpu')
@@ -130,7 +132,9 @@ def get_pred(name, img):
 def app():
     prepare_data()
 
-    model_name = st.sidebar.selectbox("Model:", ['resnet50_8xb32_in1k'])
+    model_name = st.sidebar.selectbox(
+        "Model:",
+        [m.split('_prototype.pt')[0] for m in os.listdir(CACHED_PATH)])
 
     st.markdown(
         "<h1>Image To Image Retrieval</h1>",

cache/beit-base_3rdparty_in1k_prototype.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:903c5cd62888b25e31918b365e5e0141019ab21e497b6390b7a90e295e0ddf5f
+size 8998649

cache/beitv2-base_3rdparty_in1k_prototype.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e3363d51bba86a042b3a09d84fd644aa3ac2ba7df911fa2c26581ef3844440fe
+size 8998649

cache/convnext-base_32xb128_in1k_prototype.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9d9a26d889f5690cc07e52c78ef390411e95d7d794c4d236f37b43682359802b
+size 11997945

cache/mobilenet-v3-large_3rdparty_in1k_prototype.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:54e4098c4c03b509e207bebe775233e39af697f27236158e5d1145ded7434dd3
+size 11248121