Project information

  • Category: AI & IoT Application
  • Tech Stack: Python and PyTorch Framework

Project Details

The project involves creating a custom vehicle detection algorithm based on the YOLOv5, capable of detecting four distinct classes of vehicles. This is combined with a DeepSORT object tracking algorithm to track the vehicles across different frames in a video sequence. Additionally, a custom vehicle counting method was implemented and used for counting the number of vehicles crossing a virtual polygon area in the highway in real-time.