2025-07-31 22:20 🐛

app.py

@@ -25,26 +25,26 @@ loaded_model = None
 current_model_config = {"variant": None, "dataset": None, "metric": None}
 
 pretrained_models = [
-    "ZIP-B @ ShanghaiTech A @ MAE", "ZIP-B @ ShanghaiTech A @ RMSE", "ZIP-B @ ShanghaiTech A @ NAE",
-    "ZIP-B @ ShanghaiTech B @ MAE", "ZIP-B @ ShanghaiTech B @ RMSE", "ZIP-B @ ShanghaiTech B @ NAE",
-    "ZIP-B @ UCF-QNRF @ MAE", "ZIP-B @ UCF-QNRF @ RMSE", "ZIP-B @ UCF-QNRF @ NAE",
-    "ZIP-B @ NWPU-Crowd @ MAE", "ZIP-B @ NWPU-Crowd @ RMSE", "ZIP-B @ NWPU-Crowd @ NAE",
+    "ZIP-B @ ShanghaiTech A @ MAE", "ZIP-B @ ShanghaiTech A @ NAE",
+    "ZIP-B @ ShanghaiTech B @ MAE", "ZIP-B @ ShanghaiTech B @ NAE",
+    "ZIP-B @ UCF-QNRF @ MAE", "ZIP-B @ UCF-QNRF @ NAE",
+    "ZIP-B @ NWPU-Crowd @ MAE", "ZIP-B @ NWPU-Crowd @ NAE",
     "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━",
-    "ZIP-S @ ShanghaiTech A @ MAE", "ZIP-S @ ShanghaiTech A @ RMSE", "ZIP-S @ ShanghaiTech A @ NAE",
-    "ZIP-S @ ShanghaiTech B @ MAE", "ZIP-S @ ShanghaiTech B @ RMSE", "ZIP-S @ ShanghaiTech B @ NAE",
-    "ZIP-S @ UCF-QNRF @ MAE", "ZIP-S @ UCF-QNRF @ RMSE", "ZIP-S @ UCF-QNRF @ NAE",
+    "ZIP-S @ ShanghaiTech A @ MAE", "ZIP-S @ ShanghaiTech A @ NAE",
+    "ZIP-S @ ShanghaiTech B @ MAE", "ZIP-S @ ShanghaiTech B @ NAE",
+    "ZIP-S @ UCF-QNRF @ MAE", "ZIP-S @ UCF-QNRF @ NAE",
     "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━",
-    "ZIP-T @ ShanghaiTech A @ MAE", "ZIP-T @ ShanghaiTech A @ RMSE", "ZIP-T @ ShanghaiTech A @ NAE",
-    "ZIP-T @ ShanghaiTech B @ MAE", "ZIP-T @ ShanghaiTech B @ RMSE", "ZIP-T @ ShanghaiTech B @ NAE",
-    "ZIP-T @ UCF-QNRF @ MAE", "ZIP-T @ UCF-QNRF @ RMSE", "ZIP-T @ UCF-QNRF @ NAE",
+    "ZIP-T @ ShanghaiTech A @ MAE", "ZIP-T @ ShanghaiTech A @ NAE",
+    "ZIP-T @ ShanghaiTech B @ MAE", "ZIP-T @ ShanghaiTech B @ NAE",
+    "ZIP-T @ UCF-QNRF @ MAE", "ZIP-T @ UCF-QNRF @ NAE",
     "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━",
-    "ZIP-N @ ShanghaiTech A @ MAE", "ZIP-N @ ShanghaiTech A @ RMSE", "ZIP-N @ ShanghaiTech A @ NAE",
-    "ZIP-N @ ShanghaiTech B @ MAE", "ZIP-N @ ShanghaiTech B @ RMSE", "ZIP-N @ ShanghaiTech B @ NAE",
-    "ZIP-N @ UCF-QNRF @ MAE", "ZIP-N @ UCF-QNRF @ RMSE", "ZIP-N @ UCF-QNRF @ NAE",
+    "ZIP-N @ ShanghaiTech A @ MAE", "ZIP-N @ ShanghaiTech A @ NAE",
+    "ZIP-N @ ShanghaiTech B @ MAE", "ZIP-N @ ShanghaiTech B @ NAE",
+    "ZIP-N @ UCF-QNRF @ MAE", "ZIP-N @ UCF-QNRF @ NAE",
     "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━",
-    "ZIP-P @ ShanghaiTech A @ MAE", "ZIP-P @ ShanghaiTech A @ RMSE", "ZIP-P @ ShanghaiTech A @ NAE",
-    "ZIP-P @ ShanghaiTech B @ MAE", "ZIP-P @ ShanghaiTech B @ RMSE", "ZIP-P @ ShanghaiTech B @ NAE",
-    "ZIP-P @ UCF-QNRF @ MAE", "ZIP-P @ UCF-QNRF @ RMSE", "ZIP-P @ UCF-QNRF @ NAE",
+    "ZIP-P @ ShanghaiTech A @ MAE", "ZIP-P @ ShanghaiTech A @ NAE",
+    "ZIP-P @ ShanghaiTech B @ MAE", "ZIP-P @ ShanghaiTech B @ NAE",
+    "ZIP-P @ UCF-QNRF @ MAE", "ZIP-P @ UCF-QNRF @ NAE",
 ]
 
 # -----------------------------
@@ -337,6 +337,10 @@ def predict(image: Image.Image, variant_dataset_metric: str):
             loaded_model.input_size = 672
         elif dataset_name == "nwpu":
             loaded_model.input_size = 672
+    elif isinstance(loaded_model.input_size, (list, tuple)):
+        loaded_model.input_size = loaded_model.input_size[0]  # Use the first element if it's a list or tuple
+    else:
+        assert isinstance(loaded_model.input_size, (int, float)), f"input_size must be an int or float, got {type(loaded_model.input_size)}"
 
     loaded_model.to(device)