Hugging Face's logo

charuka0
/
acne-multilabel-classifier

Image Classification
timm
PyTorch
multi-label-image-classification
dermatology
acne
Model card Files
xet
 Community
acne-multilabel-classifier
File size: 1,078 Bytes
2f58cf2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
{
  "task": "multi-label acne type recognition",
  "classes": [
    "Whitehead",
    "Blackhead",
    "Papule",
    "Pustule",
    "Nodule"
  ],
  "backbone": "efficientnet_b0",
  "img_size": 320,
  "normalization": {
    "mean": [
      0.485,
      0.456,
      0.406
    ],
    "std": [
      0.229,
      0.224,
      0.225
    ]
  },
  "thresholds": [
    0.9300000000000002,
    0.7200000000000002,
    0.7500000000000002,
    0.5000000000000001,
    0.8200000000000002
  ],
  "best_epoch": 20,
  "test_metrics": {
    "subset_accuracy": 0.9797979797979798,
    "label_accuracy": 0.9950937950937951,
    "hamming_loss": 0.004906204906204906,
    "macro_precision": 0.9920701754385967,
    "macro_recall": 0.9859189631845048,
    "macro_f1": 0.9889388150515815,
    "micro_precision": 0.9912790697674418,
    "micro_recall": 0.9841269841269841,
    "micro_f1": 0.9876900796524257,
    "macro_auc": 0.9992512177176863,
    "macro_ap": 0.9978351524589545
  },
  "n_train": 3231,
  "n_val": 693,
  "n_test": 693,
  "torch_version": "2.10.0+cu128",
  "timm_version": "1.0.26"
}