LoliRimuru commited on
Commit
f01bb23
·
verified ·
1 Parent(s): 81b73bb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +49 -3
README.md CHANGED
@@ -1,3 +1,49 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - image-classification
5
+ ---
6
+
7
+ # AAL-Plus Image Quality Assessment
8
+
9
+ A lightweight model that predicts compression quality levels and detects artifacts across multiple image formats (JPEG, WebP, AVIF, JXL). Trained to identify subtle compression artifacts that indicate the quality level used during image encoding.
10
+
11
+ ## 📊 Model Performance
12
+
13
+ Based on validation results from the final training epochs:
14
+
15
+ | Metric | Value |
16
+ |--------|-------|
17
+ | **Overall Validation Accuracy** | **94.2%** |
18
+ | **Overall Validation Loss** | **0.0007** |
19
+ | **Training Accuracy** | **92.7%** |
20
+
21
+ ### Per-Format Accuracy
22
+
23
+ | Format | Validation Acc | Training Acc | Quality Range |
24
+ |--------|----------------|--------------|---------------|
25
+ | **JPEG** | **99.6%** | 98.9% | 0-100 |
26
+ | **WebP** | **97.2%** | 95.7% | 0-100 |
27
+ | **AVIF** | **95.4%** | 93.3% | 0-63 |
28
+ | **JXL** | **84.7%** | 83.1% | 0-25 |
29
+
30
+ *Accuracy measured as predictions within ±5% range of actual quality values*
31
+
32
+ ## 🎯 Key Features
33
+
34
+ - **Multi-format Support**: Detects artifacts in JPEG, WebP, AVIF, and JXL formats
35
+ - **Lightweight Architecture**: Only **~2M parameters** (~8MB model size)
36
+ - **High Precision**: 99.6% on JPEG
37
+ - **Fast Inference**: Optimized for real-time processing even on very weak HW using either CPU or GPU
38
+ - **Format-agnostic**: Single model handles all compression types
39
+
40
+ ## Limitations
41
+
42
+ 1. **JXL Performance**: Lower accuracy (84.7%) due to JXL's smaller quality range (0-25) and newer compression algorithm
43
+ 2. **Resolution Dependency**: Model trained on 512x512 crops; very small images may yield reduced accuracy
44
+ 3. **Quality Range**: Model assumes standard quality ranges; non-standard encoders may produce different results
45
+
46
+ ## Environmental Impact
47
+
48
+ - **Training**: ~12 GPU hours on RTX 5090
49
+ - **Model Size**: 8MB (reduces storage and bandwidth costs)