nenzilea commited on
Commit
ecbc16d
·
verified ·
1 Parent(s): 7078477

Update readme

Browse files
Files changed (1) hide show
  1. 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 ([`nenzilea/car-classification`](https://huggingface.co/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: `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,7 +24,7 @@ This app compares 3 image classification approaches on car images:
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,7 +41,7 @@ This app compares 3 image classification approaches on car images:
41
 
42
  ## Hugging Face Space
43
 
44
- - App link: [https://huggingface.co/spaces/nenzilea/car-classification](https://huggingface.co/spaces/nenzilea/car-classification)
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