fix: 🐛 fix clipiqa model error

Author: chaofengc

pyiqa/archs/clipiqa_arch.py

@@ -139,11 +139,13 @@ def __init__(self,
         for p in self.clip_model[0].parameters():
             p.requires_grad = False
 
-        if pretrained:
+        if pretrained and 'clipiqa+' in model_type:
             if model_type == 'clipiqa+' and backbone == 'RN50':
                 self.prompt_learner.ctx.data = torch.load(load_file_from_url(default_model_urls['clipiqa+']))
-            else:
+            elif model_type in default_model_urls.keys():
                 load_pretrained_network(self, default_model_urls[model_type], True, 'params')
+            else:
+                raise(f'No pretrained model for {model_type}')
 
     def forward(self, x):
         # preprocess image

tests/IAA_benchmark_results.csv

@@ -1,3 +1,4 @@
 Metric name,ava(PLCC/SRCC/KRCC)
 nima-vgg16-ava,0.6624/0.657/0.4719
 nima,0.7172/0.7126/0.5213
+clipiqa,0.3576/0.3383/0.2301