highway-vehicle-detection

Highway Vehicle Detection Model

A YOLOv8m-based vehicle detection model trained for highway traffic monitoring and analysis.

Model Details

• Architecture: YOLOv8m (Medium)
• Training Data: 8,219 highway images with 8 vehicle classes
• Fine-tuning: Additional 92 images for improved truck/bus detection
• Classes: auto, bus, car, lcv, motorcycle, multiaxle, tractor, truck

Performance

• Stage 1: Initial training on main dataset
• Stage 2: Fine-tuned with improved dataset for better truck/bus classification
• Final Model: yolov8m_stage2_improved - Best performing model

Usage

Installation

pip install ultralytics opencv-python

Quick Start

from ultralytics import YOLO

# Load the model
model = YOLO('bichuche0705/highway-vehicle-detection')

# Run inference on an image
results = model('path/to/image.jpg')

# Display results
results[0].show()

Video Processing

# Process video with vehicle counting
results = model('path/to/video.mp4', save=True)

Model Files

• best.pt: The trained model weights (Stage 2 improved)
• data.yaml: Dataset configuration
• README.md: This model card

Dataset

The model was trained on highway traffic data with the following class distribution:

• Car: Most common vehicle type
• Truck: Heavy vehicles (improved detection in Stage 2)
• Bus: Public transport vehicles
• LCV: Light Commercial Vehicles
• Motorcycle: Two-wheelers
• Auto: Three-wheelers
• Multiaxle: Multi-axle heavy vehicles
• Tractor: Agricultural/construction vehicles

Training Process

1. Stage 1: Initial training on 8,219 images
2. Stage 2: Fine-tuning on additional 92 images focusing on truck/bus examples
3. Final Model: Combined training for optimal performance

Applications

• Highway traffic monitoring
• Vehicle counting and classification
• Traffic flow analysis
• Automated traffic surveillance
• Road safety monitoring

License

MIT License - See LICENSE file for details

Citation

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

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