ABHIMANYU PRASAD commited on
Update README.md
Browse files
README.md
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
---
|
| 2 |
-
title: Vehicle Classification Model
|
| 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
|