/Multi-type_vehicles_flow_statistics

According to YOLOv3 and SORT algorithms, counting multi-type vehicles. Implemented by Pytorch.

Primary LanguagePythonMIT LicenseMIT

Multi-type_vehicles_flow_statistics

According to YOLOv3 and SORT algorithms, counting multi-type vehicles. Implemented by Pytorch.
Detecting and tracking the vehicles in ["bicycle","bus","car","motorbike","truck"].

Reference

Dependencies

  • ubuntu/windows
  • cuda>=10.0
  • python>=3.6
  • pip3 install -r requirements.txt

Usage

  1. Download the pre-trained yolov3 weight file here and put it into weights directory;
  2. Run python3 app.py ;
  3. Select video and double click the image to select area, and then start;
  4. After detecting and tracking, the result video and file are saved under results directory, the line of results.txt with format [videoName,id,objectName] for each vehicle.

Demo

avatar

avatar

avatar