You can deploy this model and start earning money today!

vietnguyennn0705/
highway-vehicle-detection-full

Views

No views yet

Highway Vehicle Detection - Complete Project

A comprehensive vehicle detection system for highway traffic monitoring, including trained models, datasets, videos, and complete source code.

📁 Project Contents

🤖 Trained Models

runs/detect/yolov8m_stage1_smart/weights/best.pt - Stage 1 model
runs/detect/yolov8m_stage2_improved/weights/best.pt - Final improved model

📊 Datasets

dataset/ - Main training dataset (8,219 images + labels)
new_finetunedata/ - Fine-tuning dataset (92 images + labels)

🎬 Videos

input and output/test_video.mp4 - Input test video
input and output/detection_output_improved.mp4 - Output with detections

💻 Source Code

main.py - Main application for vehicle detection and counting
test_improved_model.bat - Testing script
PROJECT_REPORT.md - Complete project documentation

📋 Configuration

dataset/data.yaml - Main dataset configuration
new_finetunedata/data.yaml - Fine-tuning dataset configuration
requirements.txt - Python dependencies

🚀 Quick Start

Installation

pip install ultralytics opencv-python numpy

Run Detection

python

from ultralytics import YOLO

# Load the trained model
model = YOLO('runs/detect/yolov8m_stage2_improved/weights/best.pt')

# Process video
results = model('input and output/test_video.mp4', save=True)

Using the Main Application

python main.py

📈 Model Performance

Classes Detected

auto - Three-wheelers
bus - Public transport vehicles
car - Passenger cars
lcv - Light Commercial Vehicles
motorcycle - Two-wheelers
multiaxle - Multi-axle heavy vehicles
tractor - Agricultural/construction vehicles
truck - Heavy vehicles

Training Stages

Stage 1: Initial training on 8,219 highway images
Stage 2: Fine-tuning on 92 additional images for improved truck/bus detection

📊 Dataset Statistics

Main Dataset (`dataset/`)

Images: 8,219 highway traffic images
Labels: 8,219 corresponding annotation files
Classes: 8 vehicle types
Format: YOLO format (.txt files)

Fine-tuning Dataset (`new_finetunedata/`)

Images: 92 additional images
Focus: Improved truck and bus detection
Classes: Enhanced examples for problematic classes

🎯 Applications

Highway traffic monitoring
Vehicle counting and classification
Traffic flow analysis
Automated surveillance systems
Road safety monitoring
Traffic data collection

📁 File Structure

highway-vehicle-detection-full/
├── runs/
│   └── detect/
│       ├── yolov8m_stage1_smart/
│       │   └── weights/
│       │       ├── best.pt
│       │       └── last.pt
│       └── yolov8m_stage2_improved/
│           └── weights/
│               ├── best.pt
│               └── last.pt
├── dataset/
│   ├── data.yaml
│   └── train/
│       ├── images/ (8,219 images)
│       └── labels/ (8,219 label files)
├── new_finetunedata/
│   ├── data.yaml
│   └── train/
│       ├── images/ (92 images)
│       └── labels/ (92 label files)
├── input and output/
│   ├── test_video.mp4
│   └── detection_output_improved.mp4
├── main.py
├── PROJECT_REPORT.md
├── requirements.txt
└── README.md

🔧 Technical Details

Model Architecture: YOLOv8m (Medium)
Framework: Ultralytics YOLO
Input Format: Images/Videos
Output Format: Bounding boxes with class labels and confidence scores
Training Hardware: GPU recommended
Inference: CPU/GPU compatible

📝 Usage Examples

Basic Detection

python

from ultralytics import YOLO
import cv2

# Load model
model = YOLO('runs/detect/yolov8m_stage2_improved/weights/best.pt')

# Detect objects in image
results = model('path/to/image.jpg')

# Process results
for result in results:
    boxes = result.boxes
    for box in boxes:
        x1, y1, x2, y2 = box.xyxy[0]
        conf = box.conf[0]
        cls = int(box.cls[0])
        print(f"Class: {cls}, Confidence: {conf:.2f}")

Video Processing

python

# Process video file
results = model('input and output/test_video.mp4', save=True, save_txt=True)

📄 License

MIT License - See LICENSE file for details

🤝 Contributing

This is a research project. For questions or improvements, please contact the author.

📞 Contact

Author: Nguyen Quoc Viet
Repository: https://huggingface.co/datasets/bichuche0705/highway-vehicle-detection-full

📚 Citation

If you use this dataset or model in your research, please cite:

bibtex

@misc{highway-vehicle-detection-full,
  title={Highway Vehicle Detection - Complete Project},
  author={Nguyen Quoc Viet},
  year={2024},
  url={https://huggingface.co/datasets/bichuche0705/highway-vehicle-detection-full}
}

vietnguyennn0705/highway-vehicle-detection-full

Highway Vehicle Detection - Complete Project

📁 Project Contents

🤖 Trained Models

📊 Datasets

🎬 Videos

💻 Source Code

📋 Configuration

🚀 Quick Start

Installation

Run Detection

Using the Main Application

📈 Model Performance

Classes Detected

Training Stages

📊 Dataset Statistics

Main Dataset (dataset/)

Fine-tuning Dataset (new_finetunedata/)

🎯 Applications

📁 File Structure

🔧 Technical Details

📝 Usage Examples

Basic Detection

Video Processing

📄 License

🤝 Contributing

📞 Contact

📚 Citation

vietnguyennn0705/
highway-vehicle-detection-full

Main Dataset (`dataset/`)

Fine-tuning Dataset (`new_finetunedata/`)