Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -1,7 +1,8 @@
|
|
| 1 |
---
|
| 2 |
library_name: timm
|
| 3 |
pipeline_tag: image-classification
|
| 4 |
-
base_model:
|
|
|
|
| 5 |
tags:
|
| 6 |
- anime-classification
|
| 7 |
- real-photos
|
|
@@ -16,14 +17,11 @@ inference: true
|
|
| 16 |
|
| 17 |
# Anime/Real/Rendered Image Classifier (EfficientNet-B0)
|
| 18 |
|
| 19 |
-
|
| 20 |
-
|
| 21 |
-
**Base Model:** [google/efficientnet-b0](https://huggingface.co/google/efficientnet-b0)
|
| 22 |
|
| 23 |
## Model Details
|
| 24 |
|
| 25 |
- **Architecture:** EfficientNet-B0 (timm)
|
| 26 |
-
- **Base Model:** Google's EfficientNet-B0 (ImageNet pretrained)
|
| 27 |
- **Input Size:** 224×224 RGB
|
| 28 |
- **Classes:** anime, real, rendered
|
| 29 |
- **Parameters:** 5.3M
|
|
@@ -84,7 +82,6 @@ print(f"{labels[pred_class]}: {probs[0, pred_class]:.2%}")
|
|
| 84 |
|
| 85 |
## Training Details
|
| 86 |
|
| 87 |
-
- **Base Model:** ImageNet-pretrained EfficientNet-B0
|
| 88 |
- **Augmentation:** Raw (resize only)
|
| 89 |
- **Optimizer:** AdamW (lr=0.001)
|
| 90 |
- **Loss:** CrossEntropyLoss with class weighting
|
|
|
|
| 1 |
---
|
| 2 |
library_name: timm
|
| 3 |
pipeline_tag: image-classification
|
| 4 |
+
base_model:
|
| 5 |
+
- google/efficientnet-b0
|
| 6 |
tags:
|
| 7 |
- anime-classification
|
| 8 |
- real-photos
|
|
|
|
| 17 |
|
| 18 |
# Anime/Real/Rendered Image Classifier (EfficientNet-B0)
|
| 19 |
|
| 20 |
+
**Fast, lightweight classifier for distinguishing photographs from anime and 3D rendered images.**
|
|
|
|
|
|
|
| 21 |
|
| 22 |
## Model Details
|
| 23 |
|
| 24 |
- **Architecture:** EfficientNet-B0 (timm)
|
|
|
|
| 25 |
- **Input Size:** 224×224 RGB
|
| 26 |
- **Classes:** anime, real, rendered
|
| 27 |
- **Parameters:** 5.3M
|
|
|
|
| 82 |
|
| 83 |
## Training Details
|
| 84 |
|
|
|
|
| 85 |
- **Augmentation:** Raw (resize only)
|
| 86 |
- **Optimizer:** AdamW (lr=0.001)
|
| 87 |
- **Loss:** CrossEntropyLoss with class weighting
|