MSBT

This is a web service that recognizes wearing a helmet.
If a user captures a picture using a webcam, he or she can recognize whether he or she is wearing a helmet.

---

Members

• 박기덕(http://github.com/koreandrum97)
• 김동용(https://github.com/kingyong9169)
• 김웅수(https://github.com/CJavaPython)
• 김효진(https://github.com/hy57in)

---

Content

• 0. Preview
• 1. Technology stack
• 2. Quick Start - Docker
• 3. References

---

Preview

Demo

• Success

• Fail

Function

• App Bar

• Stepper

• Webcam ON,OFF

• Recapture

---

Technology stack

Architecture

AI

• 
• 
• 
• YOLO v5

Frontend

• 
• 
• 

Backend

• 
• WSGI

ETC

• 
• 
• 
• 

---

Quick Start - Docker

1-1. Environment Setting

• git clone

  git clone https://github.com/MSBT-2021-Silicon-Valley-Internship/Helmet-Detection.git
  cd Helmet-Detection

• Frontend : React

  # npm install
  cd client
  npm i
  
  # npm build
  npm run build

1-2. Docker compose

• docker-compose

  # make sure to be in the directory where your docker-compose.yaml
  docker-compose up --build
  docker-compose down

• Checkout web service at http://localhost:8080

---

References

Structure

|-- client
    |
    |-- build // npm build
    |
    |-- src
    |   |   |-- client
    |   |   |   |-- Root.js
    |   |   |
    |   |   |-- pages
    |   |   |   |-- Camera // camera page
    |   |   |   |-- Home // home page, stepper
    |   |   |   |-- Navigation // app bar
    |   |   |   |-- Result // result page
    |   |   |   |
    |   |   |   |-- index.js
    |   |   |
    |   |   |-- shared
    |   |   |   |-- App.js
    |   |   |
    |   |   |-- index.js
    |   |   |-- serviceWorker.js
    |
    |-- Dockerfile
    |-- package.json // npm install
    |-- container_server.conf // NGINX configuration

|-- server
    |
    |-- model // YOLO v5
    |-- utils // YOLO v5
    |
    |-- app.py // Flask server
    |-- run.py // WSGI server
    |
    |-- files.py // File upload and download
    |-- predict.py // Predict input image
    |-- yolov5s_helmet.pt // Pretrained Pytorch weights
    |
    |-- requirements.txt
    |-- Dockerfile

YOLO v5

https://github.com/ultralytics/yolov5

Swagger

https://app.swaggerhub.com/apis-docs/MSBT_2021/Helmet-Detection/1.0.0