Mitchins commited on
Commit
5c3fc05
·
verified ·
1 Parent(s): 97939e9

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +3 -6
README.md CHANGED
@@ -1,7 +1,8 @@
1
  ---
2
  library_name: timm
3
  pipeline_tag: image-classification
4
- base_model: google/efficientnet-b0
 
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
- Fine-tuned EfficientNet-B0 for distinguishing photographs from anime and 3D rendered images.
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