https://github.com/dongryeollee1/mask-detection

mask detection project through object detection to experience various model
https://github.com/dongryeollee1/mask-detection

Last synced: 3 months ago
JSON representation

mask detection project through object detection to experience various model

• Host: GitHub
• URL: https://github.com/dongryeollee1/mask-detection
• Owner: DONGRYEOLLEE1
• Created: 2022-07-25T11:13:31.000Z (almost 3 years ago)
• Default Branch: main
• Last Pushed: 2023-02-07T13:36:20.000Z (over 2 years ago)
• Last Synced: 2025-01-05T04:23:51.370Z (5 months ago)
• Language: Jupyter Notebook
• Size: 13.9 MB
• Stars: 0
• Watchers: 1
• Forks: 0
• Open Issues: 1
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# Mask Detection

## YOLOv7

[YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors](https://arxiv.org/abs/2207.02696)

## Requirements

- [LINK](https://github.com/WongKinYiu/yolov7/blob/main/requirements.txt)

```python

!pip install -r requirements.txt

```

## Weight

![weight](./image/weight_pic.PNG)

- `YOLOv7-X` 사용

## Data

- 😷[Kaggle_mask_detection](https://www.kaggle.com/datasets/andrewmvd/face-mask-detection) 사용

### Data tree

```python

root

├──annotations (folder)

├  ├── maksssksksss0.xml

├  ├── maksssksksss1.xml

├  ├── maksssksksssn.xml

├──images (folder)

├  ├── maksssksksss0.png

├  ├── maksssksksss1.png

├  ├── maksssksksssn.png

└──dd.yaml

```

### Preprocessing

> For now, preprocessed as [ROBOFLOW](https://roboflow.com/) 

> [`xml to txt`](https://github.com/Isabek/XmlToTxt)을 사용한 전처리 추가 예정

### Expected data tree when preprocessing

```python

root

├──train

├  ├──images

├  ├  ├──images01.png

├  ├──label

├  ├  ├──images01.txt

├──test

├  ├──images

├  ├  ├──images01.png

├  ├──label

├  ├  ├──images01.txt

├──valid

├  ├──images

├  ├  ├──images01.png

├  ├──label

├  ├  ├──images01.txt

├──data.yaml

```

## Performance

### Training

![train_result](./train/results.png)

![train_pred](./train/test_batch1_pred.jpg)

### Test

![test_result](./test/confusion_matrix.png)

### Inference

![inference_result](./inference/12.png)

![inference_video mp4_000000680](https://user-images.githubusercontent.com/82855597/180775979-f06621f7-26e4-4816-bd36-b687f55fafab.gif)