ecandeloro commited on
Commit
27d8f12
β€’
1 Parent(s): 2dea304

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -36,7 +36,7 @@ Siamo dei ricercatori del laboratorio [AImageLab](https://aimagelab.ing.unimore.
36
  Alcuni di noi lavorano sul **Medical Imaging con uso di Intelligenza Artificiale** πŸ§ πŸ§‘β€βš•οΈπŸ‘©β€βš•οΈπŸ₯Ό
37
 
38
  ### Technical Details πŸ€“
39
- The architecture used is a pre- trained Vision Transformer (ViT) on the ImageNet21k, with a fine-tuning on the [HAM10k dataset](https://huggingface.co/datasets/marmal88/skin_cancer) and a modified head to accommodate for the classes: Classes: Benign keratosis-like lesions, Basal cell carcinoma, Actinic keratoses, Vascular lesions, Melanocytic nevi, Melanoma, Dermatofibroma.
40
  The best validation accuracy obtained was 0.9695. However this score is not a good indicator of performance given the class imbalances present in the dataset.
41
 
42
  ### Credits
 
36
  Alcuni di noi lavorano sul **Medical Imaging con uso di Intelligenza Artificiale** πŸ§ πŸ§‘β€βš•οΈπŸ‘©β€βš•οΈπŸ₯Ό
37
 
38
  ### Technical Details πŸ€“
39
+ The architecture used is a pre- trained Vision Transformer (ViT) on the ImageNet21k, with a fine-tuning on the [HAM10k dataset](https://huggingface.co/datasets/marmal88/skin_cancer) and a modified head to accommodate for the classes: Benign keratosis-like lesions, Basal cell carcinoma, Actinic keratoses, Vascular lesions, Melanocytic nevi, Melanoma, Dermatofibroma.
40
  The best validation accuracy obtained was 0.9695. However this score is not a good indicator of performance given the class imbalances present in the dataset.
41
 
42
  ### Credits