Spaces:
Sleeping
Sleeping
Update readme
Browse files- readme_.md +4 -4
readme_.md
CHANGED
|
@@ -2,13 +2,13 @@
|
|
| 2 |
|
| 3 |
This app compares 3 image classification approaches on car images:
|
| 4 |
|
| 5 |
-
- Fine-tuned ViT model (
|
| 6 |
- Zero-shot CLIP (`openai/clip-vit-large-patch14`)
|
| 7 |
- OpenAI vision model (GPT-4o image classification)
|
| 8 |
|
| 9 |
## Dataset Used For Training
|
| 10 |
|
| 11 |
-
- Hugging Face dataset:
|
| 12 |
- The Stanford Cars dataset contains 196 fine-grained classes (car make/model/year combinations). We group them into 9 brand-level classes for a cleaner, more visually meaningful classification task.
|
| 13 |
- Number of classes: `9`
|
| 14 |
- Classes: `BMW`, `Dodge`, `Ferrari`, `Ford`, `Jeep`, `Lamborghini`, `Porsche`, `Rolls-Royce`, `Toyota`
|
|
@@ -24,7 +24,7 @@ This app compares 3 image classification approaches on car images:
|
|
| 24 |
|
| 25 |
## Trained Model
|
| 26 |
|
| 27 |
-
- Hugging Face model link:
|
| 28 |
- Base model: `google/vit-base-patch16-224`
|
| 29 |
- Only the final classification head was fine-tuned (all other layers frozen).
|
| 30 |
- Trainable parameters: ~4,614 out of ~85.8M total.
|
|
@@ -41,7 +41,7 @@ This app compares 3 image classification approaches on car images:
|
|
| 41 |
|
| 42 |
## Hugging Face Space
|
| 43 |
|
| 44 |
-
- App link:
|
| 45 |
|
| 46 |
## Example Image Results
|
| 47 |
|
|
|
|
| 2 |
|
| 3 |
This app compares 3 image classification approaches on car images:
|
| 4 |
|
| 5 |
+
- Fine-tuned ViT model (`nenzilea/car-classification`)
|
| 6 |
- Zero-shot CLIP (`openai/clip-vit-large-patch14`)
|
| 7 |
- OpenAI vision model (GPT-4o image classification)
|
| 8 |
|
| 9 |
## Dataset Used For Training
|
| 10 |
|
| 11 |
+
- Hugging Face dataset: https://huggingface.co/datasets/tanganke/stanford_cars
|
| 12 |
- The Stanford Cars dataset contains 196 fine-grained classes (car make/model/year combinations). We group them into 9 brand-level classes for a cleaner, more visually meaningful classification task.
|
| 13 |
- Number of classes: `9`
|
| 14 |
- Classes: `BMW`, `Dodge`, `Ferrari`, `Ford`, `Jeep`, `Lamborghini`, `Porsche`, `Rolls-Royce`, `Toyota`
|
|
|
|
| 24 |
|
| 25 |
## Trained Model
|
| 26 |
|
| 27 |
+
- Hugging Face model link: https://huggingface.co/nenzilea/car-classification
|
| 28 |
- Base model: `google/vit-base-patch16-224`
|
| 29 |
- Only the final classification head was fine-tuned (all other layers frozen).
|
| 30 |
- Trainable parameters: ~4,614 out of ~85.8M total.
|
|
|
|
| 41 |
|
| 42 |
## Hugging Face Space
|
| 43 |
|
| 44 |
+
- App link: https://huggingface.co/spaces/nenzilea/car-classification
|
| 45 |
|
| 46 |
## Example Image Results
|
| 47 |
|