Update README.md
Browse files
README.md
CHANGED
|
@@ -1,146 +1,37 @@
|
|
| 1 |
-
# π§ SmartVision AI
|
| 2 |
-
### Intelligent Multi-Class Object Recognition System
|
| 3 |
-
|
| 4 |
-
SmartVision AI is an **end-to-end Computer Vision application** that performs
|
| 5 |
-
**image classification**, **object detection**, and **real-time inference** using
|
| 6 |
-
state-of-the-art deep learning models.
|
| 7 |
-
The project demonstrates the complete AI lifecycle β from **model training** to
|
| 8 |
-
**optimized deployment** using **Streamlit**.
|
| 9 |
-
|
| 10 |
---
|
| 11 |
-
|
| 12 |
-
|
| 13 |
-
|
| 14 |
-
|
| 15 |
-
|
| 16 |
-
|
| 17 |
-
|
| 18 |
-
|
| 19 |
-
- π¦ **Object Detection**
|
| 20 |
-
- Pretrained YOLOv8 model for real-time object detection
|
| 21 |
-
- Bounding boxes, class labels, and confidence scores
|
| 22 |
-
- Adjustable confidence threshold
|
| 23 |
-
|
| 24 |
-
- πΈ **Live Webcam Detection (Optimized)**
|
| 25 |
-
- Real-time detection using webcam
|
| 26 |
-
- FPS monitoring and CPU-friendly optimizations
|
| 27 |
-
- Frame skipping and resolution scaling
|
| 28 |
-
|
| 29 |
-
- π **Model Performance Dashboard**
|
| 30 |
-
- Accuracy comparison (Train / Validation / Test)
|
| 31 |
-
- Inference speed analysis
|
| 32 |
-
- Visual performance insights
|
| 33 |
-
|
| 34 |
-
- β‘ **Optimized Inference**
|
| 35 |
-
- Lightweight models for CPU execution
|
| 36 |
-
- Streamlit caching for faster loading
|
| 37 |
-
- Performance-focused design decisions
|
| 38 |
-
|
| 39 |
---
|
| 40 |
|
| 41 |
-
|
| 42 |
|
| 43 |
-
|
| 44 |
-
|
| 45 |
-
- **ResNet50**
|
| 46 |
-
- **MobileNetV2**
|
| 47 |
-
- **EfficientNet-B0**
|
| 48 |
|
| 49 |
-
|
| 50 |
-
-
|
| 51 |
-
|
| 52 |
-
|
|
|
|
|
|
|
| 53 |
|
| 54 |
-
##
|
| 55 |
-
|
| 56 |
-
- **Image Classification Dataset**
|
| 57 |
-
- Domain-specific dataset
|
| 58 |
-
- 25 object classes
|
| 59 |
-
- Train / Validation / Test split
|
| 60 |
-
- Image preprocessing and augmentation applied
|
| 61 |
-
|
| 62 |
-
- **Object Detection Dataset**
|
| 63 |
-
- COCO Dataset
|
| 64 |
-
- 80 general-purpose object classes
|
| 65 |
-
- Bounding box annotations
|
| 66 |
-
|
| 67 |
-
---
|
| 68 |
-
|
| 69 |
-
## π οΈ Tech Stack
|
| 70 |
-
|
| 71 |
-
**Programming Language**
|
| 72 |
-
- Python π
|
| 73 |
-
|
| 74 |
-
**Deep Learning & Computer Vision**
|
| 75 |
- PyTorch
|
| 76 |
-
- Torchvision
|
| 77 |
- Ultralytics YOLOv8
|
| 78 |
-
- OpenCV
|
| 79 |
-
|
| 80 |
-
**Data Analysis & Visualization**
|
| 81 |
-
- NumPy
|
| 82 |
-
- Pandas
|
| 83 |
-
- Matplotlib
|
| 84 |
-
- Seaborn
|
| 85 |
-
|
| 86 |
-
**Web & Deployment**
|
| 87 |
- Streamlit
|
| 88 |
-
-
|
| 89 |
-
-
|
| 90 |
-
|
| 91 |
-
---
|
| 92 |
-
|
| 93 |
-
## β‘ Performance Optimization Techniques
|
| 94 |
|
| 95 |
-
|
| 96 |
-
|
| 97 |
-
- Lightweight YOLOv8n model for CPU execution
|
| 98 |
-
- Streamlit resource caching
|
| 99 |
-
- Confidence-based filtering of predictions
|
| 100 |
|
| 101 |
---
|
| 102 |
|
| 103 |
-
|
| 104 |
-
SmartVisionAI/
|
| 105 |
-
β
|
| 106 |
-
βββ app.py # Main Streamlit application
|
| 107 |
-
βββ requirements.txt # Python dependencies
|
| 108 |
-
βββ README.md # Project documentation
|
| 109 |
-
βββ Image.txt/ # Images, icons, logos
|
| 110 |
-
βββ yolo.ipynb
|
| 111 |
-
βββ smartvisionAI.ipynb(Downloading and training process of data)
|
| 112 |
-
|
| 113 |
-
## Screenshots
|
| 114 |
-
<img width="1920" height="1080" alt="Screenshot (176)" src="https://github.com/user-attachments/assets/feb97730-b862-4504-ac37-bc733fe21aba" />
|
| 115 |
-
|
| 116 |
-
<img width="1920" height="1080" alt="Screenshot (178)" src="https://github.com/user-attachments/assets/60abc70b-3d2c-4aee-aaf2-53dade77d7e3" />
|
| 117 |
-
Demo Images
|
| 118 |
-
|
| 119 |
-
<img width="1920" height="1080" alt="Screenshot (177)" src="https://github.com/user-attachments/assets/5c03e4b4-eaa5-4eb8-942f-b3fae16db210" />
|
| 120 |
-
Detection
|
| 121 |
-
|
| 122 |
-
<img width="1253" height="825" alt="Screenshot 2025-12-14 at 08-17-06 SmartVision AI - Intelligent Multi-Class Object Recognition System" src="https://github.com/user-attachments/assets/158bdfca-c160-4968-a508-d3cd47878768" />
|
| 123 |
-
|
| 124 |
-
|
| 125 |
-
|
| 126 |
-
|
| 127 |
-
## π Note on Model Files
|
| 128 |
-
|
| 129 |
-
-Due to size constraints, trained model weights (.pt, .pth) are not included
|
| 130 |
-
in this repository.
|
| 131 |
-
|
| 132 |
-
## π Academic & Practical Relevance
|
| 133 |
-
|
| 134 |
-
- This project was built to:
|
| 135 |
-
|
| 136 |
-
- Demonstrate practical Deep Learning & Computer Vision skills
|
| 137 |
-
|
| 138 |
-
- Showcase model deployment and optimization
|
| 139 |
-
|
| 140 |
-
- Serve as a portfolio project for interviews and evaluations
|
| 141 |
-
|
| 142 |
-
## π¨βπ» Developer
|
| 143 |
-
|
| 144 |
-
Rahul Kumar
|
| 145 |
-
B.Tech in Information Technology
|
| 146 |
-
IIEST Shibpur
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
+
title: SmartVision AI
|
| 3 |
+
emoji: ποΈ
|
| 4 |
+
colorFrom: indigo
|
| 5 |
+
colorTo: blue
|
| 6 |
+
sdk: streamlit
|
| 7 |
+
sdk_version: "1.32.0"
|
| 8 |
+
app_file: src/streamlit_app.py
|
| 9 |
+
pinned: false
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 10 |
---
|
| 11 |
|
| 12 |
+
# SmartVision AI ποΈ
|
| 13 |
|
| 14 |
+
**SmartVision AI** is an intelligent multi-class object recognition system built using
|
| 15 |
+
Deep Learning and Computer Vision techniques.
|
|
|
|
|
|
|
|
|
|
| 16 |
|
| 17 |
+
## π Features
|
| 18 |
+
- Image Classification using CNN models (VGG16, ResNet50, MobileNetV2, EfficientNet-B0)
|
| 19 |
+
- Object Detection using YOLOv8
|
| 20 |
+
- Model performance comparison dashboard
|
| 21 |
+
- Optimized inference for deployment
|
| 22 |
+
- Streamlit-based interactive UI
|
| 23 |
|
| 24 |
+
## π Tech Stack
|
| 25 |
+
- Python
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 26 |
- PyTorch
|
|
|
|
| 27 |
- Ultralytics YOLOv8
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 28 |
- Streamlit
|
| 29 |
+
- OpenCV
|
| 30 |
+
- NumPy, Pandas, Matplotlib
|
|
|
|
|
|
|
|
|
|
|
|
|
| 31 |
|
| 32 |
+
## π¦ Deployment
|
| 33 |
+
This app is deployed using **Hugging Face Spaces (Streamlit SDK)**.
|
|
|
|
|
|
|
|
|
|
| 34 |
|
| 35 |
---
|
| 36 |
|
| 37 |
+
π¨βπ» Developed by **Rahul Kumar**
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|