ABHIMANYU PRASAD commited on
Commit
5fa7581
Β·
verified Β·
1 Parent(s): 0bae8cc

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +60 -2
README.md CHANGED
@@ -1,5 +1,5 @@
1
  ---
2
- title: Vehicle Classification Model-utd
3
  emoji: 🐨
4
  colorFrom: yellow
5
  colorTo: red
@@ -8,7 +8,65 @@ sdk_version: 6.9.0
8
  app_file: app.py
9
  pinned: false
10
  license: apache-2.0
11
- short_description: A CNN for vehicle perception with 78.54 accuracy.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
12
  ---
13
 
14
  Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
1
  ---
2
+ title: Vehicle Classification Model
3
  emoji: 🐨
4
  colorFrom: yellow
5
  colorTo: red
 
8
  app_file: app.py
9
  pinned: false
10
  license: apache-2.0
11
+ short_description: A CNN for vehicle perception with 78.54% accuracy.
12
+
13
+ ---
14
+ title: Vehicle Classification Demo - UTD
15
+ emoji: 🏎️
16
+ colorFrom: blue
17
+ colorTo: indigo
18
+ sdk: gradio
19
+ sdk_version: 5.16.0
20
+ app_file: app.py
21
+ pinned: true
22
+ license: apache-2.0
23
+ short_description: Custom CNN for Autonomous Vehicle Perception
24
+ ---
25
+
26
+ # 🏎️ Vehicle Classification: Autonomous Perception Demo
27
+ This application provides a real-time inference interface for a custom-built CNN architecture designed for safety-critical vehicle classification.
28
+
29
+ ## πŸ“Š Model Performance
30
+ * **Final Test Accuracy:** 78.54% (Target: >50%)
31
+ * **Generalization Gap:** 0.06% (Minimal variance between training and test sets)
32
+ * **Classes:** 8 (Bicycle, Bus, Car, Motorcycle, NonVehicles, Taxi, Truck, Van)
33
+
34
+ ## πŸ—οΈ Technical Architecture
35
+ Unlike generic transfer learning, this model utilizes a **Custom CNN** designed for high transparency:
36
+ * **Architecture:** 3-stage Convolutional blocks (16, 32, 64 filters).
37
+ * **Safety Regularization:** Integrated **Dropout (0.5)** and **BatchNorm2d** to ensure the model learns global geometric features rather than overfitting on pixel-level noise.
38
+ * **Preprocessing:** Standardized 224x224 input with ImageNet-standard normalization.
39
+
40
+ ## πŸ›‘οΈ AI Safety Analysis
41
+ * **Critical Detection:** The model is highly optimized to distinguish `NonVehicles` (1,775/1,800 correct), minimizing the risk of "false positive" emergency braking events.
42
+ * **Robustness:** Maintained high performance on the test set despite diverse lighting and perspective variations in the original dataset.
43
+
44
+ ## πŸ“š More Work & Research Portfolio
45
+ *This project is part of my broader commitment to AI/CS research and open-source development.*
46
+
47
+ ## πŸ› οΈ Key Projects
48
+ * **Language Datasets**: Curated and published 100k+ rows datasets for low resource languages on Hugging Face for regional NLP research.
49
+ * **CNSD Model Architecture**: Authored research on neural network configurations for optimized feature extraction.
50
+ * Custom 4 sentiment models, 1 vehicle classification model and several datasets.
51
+ * Link to my Hugging Face account : https://huggingface.co/abhiprd20 (with all models and datasets)
52
+ * My NLP research paper pre-print : https://zenodo.org/records/19054785
53
+ * 2nd research project : https://github.com/abhiprd200/CNSD_prototype
54
+ * **Demo of this project:** [Live on Hugging Face Spaces]((https://huggingface.co/spaces/abhiprd20/vehicle_classification_model-utd))
55
+ * My github with other projects : https://github.com/abhiprd200
56
+
57
+
58
+ ## πŸ”— Links for this project
59
+ * **[GitHub Repository](https://github.com/abhiprd200/vehicle_classification_model-utd)**
60
+ * **[Hugging Face Model Card](https://huggingface.co/abhiprd20/vehicle_classification_model-utd)**
61
+ * **[Hugging Face Spaces Deployment](https://huggingface.co/spaces/abhiprd20/vehicle_classification_model-utd)**
62
+
63
+
64
+ ## Contact
65
+ * E-mail : abhiprd20@gmail.com
66
+
67
+ ---
68
+ **License:** Apache 2.0
69
+
70
  ---
71
 
72
  Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference