File size: 1,531 Bytes
a0775f9 c4dea26 a0775f9 6ff9ad4 c4dea26 6ff9ad4 c4dea26 6ff9ad4 c4dea26 a0775f9 c4dea26 a0775f9 c4dea26 |
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 |
---
license: mit
tags:
- image-classification
- resnet50
- medical
- acne-detection
task:
- image-classification
output:
- label: "level1"
score: 0.98
widget:
- text: "example_image.jpg"
output:
- label: "level3"
score: 0.85
---
# ResNet-50 Model for Acne Severity Classification
This is a fine-tuned ResNet-50 model designed to classify the severity of acne from medical images into five categories (Severity 1 to Severity 5). The model leverages transfer learning on ResNet-50 pre-trained on ImageNet and adapts it for acne severity classification tasks.
## Model Details
### Training Details
- **Framework:** PyTorch
- **Base Model:** ResNet-50 (pretrained on ImageNet)
- **Dataset:** A balanced dataset of acne images annotated with severity levels (Severity 1 to 5).
- **Preprocessing:** Images resized to 224x224 pixels, normalized using ImageNet statistics (mean: `[0.485, 0.456, 0.406]`, std: `[0.229, 0.224, 0.225]`).
- **Optimizer:** Adam with a learning rate of 0.001.
- **Loss Function:** CrossEntropyLoss.
- **Epochs:** 10.
- **Validation Accuracy:** 0.85 (on a held-out validation set).
## Intended Use
This model is intended for educational purposes and demonstrates image classification for medical images. It should not be used for clinical decision-making without further validation.
## Example Usage
You can use this model via the Hugging Face Transformers pipeline for inference. Ensure you have the `transformers` library installed:
```bash
pip install transformers |