nadakandrew commited on
Commit
423ef09
·
verified ·
1 Parent(s): 435796c

Add README.md

Browse files
Files changed (1) hide show
  1. README.md +24 -0
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
+ ```