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 (TF-EfficientNetV2-S)
|
| 18 |
|
| 19 |
-
|
| 20 |
-
|
| 21 |
-
**Base Model:** [google/efficientnetv2-s](https://huggingface.co/google/efficientnetv2-s) (TensorFlow → PyTorch)
|
| 22 |
|
| 23 |
## Model Details
|
| 24 |
|
| 25 |
- **Architecture:** TF-EfficientNetV2-S (timm)
|
| 26 |
-
- **Base Model:** Google's EfficientNetV2-S (TensorFlow, ported to PyTorch)
|
| 27 |
- **Input Size:** 224×224 RGB
|
| 28 |
- **Classes:** anime, real, rendered
|
| 29 |
- **Parameters:** 21.5M (4× larger than B0)
|
|
@@ -96,7 +94,6 @@ print(f"{labels[pred_class]}: {probs[0, pred_class]:.2%}")
|
|
| 96 |
|
| 97 |
## Training Details
|
| 98 |
|
| 99 |
-
- **Base Model:** TensorFlow EfficientNetV2-S (ported to PyTorch via TIMM)
|
| 100 |
- **Augmentation:** Raw (resize only)
|
| 101 |
- **Optimizer:** AdamW (lr=0.001)
|
| 102 |
- **Loss:** CrossEntropyLoss with class weighting
|
|
|
|
| 1 |
---
|
| 2 |
library_name: timm
|
| 3 |
pipeline_tag: image-classification
|
| 4 |
+
base_model:
|
| 5 |
+
- google/efficientnetv2-s
|
| 6 |
tags:
|
| 7 |
- anime-classification
|
| 8 |
- real-photos
|
|
|
|
| 17 |
|
| 18 |
# Anime/Real/Rendered Image Classifier (TF-EfficientNetV2-S)
|
| 19 |
|
| 20 |
+
**Higher-capacity classifier with improved generalization for anime, photo, and 3D detection.**
|
|
|
|
|
|
|
| 21 |
|
| 22 |
## Model Details
|
| 23 |
|
| 24 |
- **Architecture:** TF-EfficientNetV2-S (timm)
|
|
|
|
| 25 |
- **Input Size:** 224×224 RGB
|
| 26 |
- **Classes:** anime, real, rendered
|
| 27 |
- **Parameters:** 21.5M (4× larger than B0)
|
|
|
|
| 94 |
|
| 95 |
## Training Details
|
| 96 |
|
|
|
|
| 97 |
- **Augmentation:** Raw (resize only)
|
| 98 |
- **Optimizer:** AdamW (lr=0.001)
|
| 99 |
- **Loss:** CrossEntropyLoss with class weighting
|