Giter VIP home page Giter VIP logo

jadhaddad92 / covid-mask-detector Goto Github PK

View Code? Open in Web Editor NEW
110.0 4.0 36.0 24.65 MB

Detect whether a person is wearing a mask or not

Home Page: https://towardsdatascience.com/how-i-built-a-face-mask-detector-for-covid-19-using-pytorch-lightning-67eb3752fd61

License: MIT License

Python 93.59% Dockerfile 6.41%
mask covid19 deep-learning detection pytorch coronavirus pytorch-lightning classification

covid-mask-detector's Introduction

covid-mask-detector's People

Contributors

davidkatz-il avatar dependabot[bot] avatar jadhaddad92 avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar  avatar  avatar  avatar

covid-mask-detector's Issues

data_preparation.py fills RAM

Describe the bug
When I run data_preparation.py on my console whit python data_preparation.py comand it start to work, but when is unzip the folder non_mask_photos something happend that stop to do it and show on the console the error: view on the screenshot
To Reproduce
Steps to reproduce the behavior:

  1. Go to cmd
  2. type python data_preparation.py
  3. wait a few minutes
  4. see the error

Expected behavior
My expected behavior was that the code unzip all the data complete and show me a doc whit the dataset whit the id (0 or 1) (mask or not)

Screenshots
error

Desktop (please complete the following information):

  • OS: Windows 10 pro 1909
  • python 3.8.1

[BUG] Model fails to load

Describe the bug
Running video.py with the provided model ('face_mask.ckpt') with any video, fails to load the model state_dict with KeyError: 'state_dict'

To Reproduce
Steps to reproduce the behavior:
Download repo and create a virtualenv(in PyCharm) from requirements.txt.
PyTorch, Pandas and PyTorch Lightning do not install out of the box, they need to be installed manually because of version errors.

Expected behavior
Output a video with box overlays over faces with or without masks.

Screenshots
image

Desktop (please complete the following information):

  • OS: Windows 10
  • Python Version 3.7.7
  • PyTorch 1.5 last stable version

Cannot train on multiGPU

Great work on Pytorch lightning, but I cannot train it on multi GPU(I set gpu=2), since Panda cannot support

[BUG] Faces not detected

Good Job with high accuracy for single person, but the accuracy lowers down for videos on the internet. I tested it with the video of Trump and his partners wearing masks. It detects very few faces and only in few frames. You can see the result Here. I also used a high resolution version of the video but nothing changed. What is the problem and how to resolve?

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.