Upload Image Quality Fusion model

README.md

@@ -0,0 +1,16 @@
+---
+license: mit
+tags:
+- aesthetic
+- brisque
+- clip
+- fusion
+- image-quality
+- model_hub_mixin
+- pytorch_model_hub_mixin
+---
+
+This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
+- Code: https://github.com/yourusername/image-quality-fusion
+- Paper: [More Information Needed]
+- Docs: [More Information Needed]

config.json

@@ -0,0 +1,11 @@
+{
+  "clip_embed_dim": 512,
+  "clip_model": "ViT-B-32",
+  "dropout": 0.3,
+  "hidden_dim": 256,
+  "model_type": "deep",
+  "output_range": [
+    1.0,
+    10.0
+  ]
+}

model_info.json

@@ -0,0 +1,21 @@
+{
+  "name": "Image Quality Fusion",
+  "version": "1.0.0",
+  "description": "Multi-modal image quality assessment combining BRISQUE, Aesthetic, and CLIP features",
+  "performance": {
+    "mse": 2.859546184539795,
+    "mae": 1.4086108207702637,
+    "r2": 0.2890252470970154
+  },
+  "training_data": "SPAQ dataset (11,125 smartphone images)",
+  "training_metadata": {
+    "epoch": 4,
+    "val_loss": 2.880443743297032,
+    "metrics": {
+      "mse": 2.859546184539795,
+      "mae": 1.4086108207702637,
+      "r2": 0.2890252470970154
+    },
+    "timestamp": "2025-08-01T03:40:12.719814"
+  }
+}