Update README.md
Browse files
README.md
CHANGED
|
@@ -41,14 +41,6 @@ model-index:
|
|
| 41 |
|
| 42 |
A PyTorch-based deep learning model for classifying skincare diseases and conditions from images using transfer learning with EfficientNet-B0, ResNet50, and Vision Transformer (ViT).
|
| 43 |
|
| 44 |
-
## Model Details
|
| 45 |
-
|
| 46 |
-
- **Architecture**: EfficientNet-B0 with transfer learning from ImageNet
|
| 47 |
-
- **Framework**: PyTorch
|
| 48 |
-
- **Input Size**: 224×224 RGB images
|
| 49 |
-
- **Classes**: Multiple skin disease categories (acne, eczema, normal, rosacea, etc.) or [classes](data_classes.txt)
|
| 50 |
-
- **Training Dataset**: [0xnu/skincare](https://huggingface.co/datasets/0xnu/skincare)
|
| 51 |
-
|
| 52 |
### Code Example
|
| 53 |
|
| 54 |
```python
|
|
|
|
| 41 |
|
| 42 |
A PyTorch-based deep learning model for classifying skincare diseases and conditions from images using transfer learning with EfficientNet-B0, ResNet50, and Vision Transformer (ViT).
|
| 43 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 44 |
### Code Example
|
| 45 |
|
| 46 |
```python
|