traffic-detection

disc: This is a application forked by roboflow/supervision (roboflow/supervision/traffic_analysis/examples)

Transit detection is a project that use IA to detect, track and count vehicles

output_result_multiples_vehicles.mov

This modified script can count, track, annotate, classify, and distinguish different classes of vehicles.

💻 install

• clone repository and navigate to example directory

  git clone https://github.com/hugoles/supervision.git
  cd supervision/examples/traffic_analysis

• setup python environment and activate it [optional]

  python3 -m venv venv
  source venv/bin/activate

• install required dependencies

  pip install -r requirements.txt

• download model4.pt and output.mov files

  ./setup.sh

⚙️ run

python script.py \
--source_weights_path data/model4.pt \
--source_video_path data/output.mov \
--confidence_threshold 0.3 \
--iou_threshold 0.5 \
--target_video_path data/output_result.mov