Add README.md
Browse files
README.md
ADDED
|
@@ -0,0 +1,24 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Image AutoML with AutoGluon, Sign Identification
|
| 2 |
+
|
| 3 |
+
Dataset: ecopus/sign_identification
|
| 4 |
+
|
| 5 |
+
Task: multiclass image classification using AutoGluon Multimodal.
|
| 6 |
+
|
| 7 |
+
## Training
|
| 8 |
+
- Preset: medium_quality
|
| 9 |
+
- Backbone: timm_image, resnet18
|
| 10 |
+
- Split: augmented 80 percent train, 20 percent test
|
| 11 |
+
- External validation: original split
|
| 12 |
+
|
| 13 |
+
## Results
|
| 14 |
+
- Augmented test accuracy: 1.0000
|
| 15 |
+
- Augmented test weighted F1: 1.0000
|
| 16 |
+
- Original external accuracy: 1.0000
|
| 17 |
+
- Original external weighted F1: 1.0000
|
| 18 |
+
|
| 19 |
+
## How to load
|
| 20 |
+
```python
|
| 21 |
+
from autogluon.multimodal import MultiModalPredictor
|
| 22 |
+
predictor = MultiModalPredictor.load('predictor_dir')
|
| 23 |
+
pred = predictor.predict({'image': ['/path/to/img.png']})
|
| 24 |
+
```
|