highway-vehicle-detection-code / TEAM_INSTRUCTIONS.md
Nguyễn Quốc Việt
Upload TEAM_INSTRUCTIONS.md with huggingface_hub
456efca verified
|
Raw
History Blame
1.55 kB

Highway Vehicle Detection - Team Instructions

Quick Setup

  1. Clone the repository:

    git clone https://huggingface.co/bichuche0705/highway-vehicle-detection-code
    cd highway-vehicle-detection-code
    
  2. Install dependencies:

    pip install ultralytics opencv-python numpy
    
  3. Run the application:

    python main.py
    

External Resources

Test Video

Watch the model in action: YouTube Demo

Main Dataset

Download the complete training dataset: Kaggle Dataset

What's Included

  • Trained Models: Ready-to-use YOLOv8m models
  • Source Code: Complete detection application
  • Fine-tuning Dataset: 92 images used for model improvement
  • Training Logs: Performance metrics and visualizations
  • Documentation: Complete project documentation

Model Files

  • models/yolov8m_stage2_improved_best.pt - Final model (recommended)
  • models/yolov8m_stage1_smart_best.pt - Stage 1 model

Usage Examples

Basic Detection

from ultralytics import YOLO
model = YOLO('models/yolov8m_stage2_improved_best.pt')
results = model('image.jpg')
results[0].show()

Video Processing

results = model('video.mp4', save=True)

Support

For questions or issues, contact the project author.