This project leverages a fine-tuned YOLO model for vehicle detection and tracking, enabling automated counting of various vehicle types as they cross a designated line using a single camera or video feed. By replacing manual tracking methods with advanced computer vision techniques, the system provides accurate real-time data on traffic flow, which can be used to inform transportation control decisions.
- 🚙 Vehicle Detection: Detects multiple types of vehicles (e.g., cars, trucks, motorcycles).
- 🔢 Vehicle Counting: Counts vehicles as they cross a specific line in the frame.
- 📊 Classification: Classifies vehicles into predefined categories.
- data/: Dataset configurations and preprocessing scripts
- models/: YOLO model configuration and weights
- notebooks/: Jupyter notebooks for model training and testing
- streamlit_app/: App with streamlit
- README.md: Project overview and usage instructions
-
Clone the repository:
git clone https://github.com/angelTCC/VehicleTracking.git
-
Install Dependencies: bash pip install -r requirements.txt
cd VehicleTracking python3.10 -m venv .env pip install -r requeriments.txt -
Download YOLO Weights: Download pretrained weights and place them in the
models/directory. -
Run the application:
cd streamlit_app streamlit run Home.py
Contributions are welcome! Please fork the repository and submit a pull request for any enhancements or bug fixes.
This project is licensed under the MIT License.
You can replace your-username with your actual GitHub username and adjust paths or descriptions as needed. This structure is ready to paste directly into a README.md file for a polished, professional look! ChatGPT said: ChatGPT