peterpaulorange / Social-Distancing-using-YOLOv5

Classifying people as high risk and low risk based on their distance to other people.

Social-Distancing-using-YOLOv5

Classifying people as high risk and low risk based on their distance to other people 😷

The original YOLOv5 model was used to detect people. Thereafter the people were classified as "High Risk" and "Low Risk" based on their distance from one another.

---

Requirements

Python 3.7 or later with all requirements.txt dependencies installed, including torch >= 1.5. To install run:

$ pip install -U -r requirements.txt

---

Usage

Run on a single video file

$ python detect.py --source ./inference/videos/videofile.mp4

Run on all video files inside a folder

$ python detect.py --source ./inference/videos/

Options

Add using '--'

agnostic_nms=False, augment=False, classes=None, conf_thres=0.4, device='', fourcc='mp4v', half=False, img_size=640, iou_thres=0.5, output='inference/output', save_txt=False, source='./inference/images/', view_img=False, weights='yolov5s.pt'

---

Output

---

TODO

• Remove a bug which classifies people as low risk instead of high.
• Add live video support
• Add image support