Introvertise

Walkthrough

The YOLOv5 model is used to detect people after which, they are classified as "Risky" and "Safe" based on their distance from one another.

Dependencies

• Python3.7+
• PyTorch >=1.5

Setup

• Install requirements

  $ pip install -U -r requirements.txt
  

Usage

• Run on a video file

  $ python detect.py --source <path_to_video_file>

• Run on webcam

  $ python detect.py --source 0

Output