# Social-Distancing-using-YOLOv5

**Repository Path**: hsulee/Social-Distancing-using-YOLOv5

## Basic Information

- **Project Name**: Social-Distancing-using-YOLOv5
- **Description**: No description available
- **Primary Language**: Unknown
- **License**: GPL-3.0
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2021-03-28
- **Last Updated**: 2021-11-13

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# Social-Distancing-using-YOLOv5



### Classifying people as high risk and low risk based on their distance to other people :mask:


The original [YOLOv5](https://github.com/ultralytics/yolov5 "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:
```bash
$ pip install -U -r requirements.txt
```


---



## Usage


Run on a single video file


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


Run on all video files inside a folder


```bash
$ 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

![](output.gif)


---


### TODO

- [x] Remove a bug which classifies people as low risk instead of high.
- [x] Add live video support
- [x] Add image support