Face Mask Detection system based on computer vision and deep learning using OpenCV and Tensorflow/Keras

Overview

Face Mask Detection

Face Mask Detection System built with OpenCV, Keras/TensorFlow using Deep Learning and Computer Vision concepts in order to detect face masks in static images as well as in real-time video streams.

                               Python contributions welcome Forks Stargazers Issues LinkedIn

                                    Live Demo

πŸ‘‡ Support me here!

Buy Me A Coffee

πŸ˜‡ Motivation

Amid the ongoing COVID-19 pandemic, there are no efficient face mask detection applications which are now in high demand for transportation means, densely populated areas, residential districts, large-scale manufacturers and other enterprises to ensure safety. The absence of large datasets of β€˜with_mask’ images has made this task cumbersome and challenging.

PPT and Project Report sharing costs β‚Ή1000 ($15)

If interested, contact me at [email protected]

🌟 Purchase at a Discounted Rate

βŒ› Project Demo

πŸŽ₯ YouTube Demo Link

πŸ’» Dev Link

Already deployed version

⚠️ TechStack/framework used

⭐ Features

Our face mask detector doesn't use any morphed masked images dataset and the model is accurate. Owing to the use of MobileNetV2 architecture, it is computationally efficient, thus making it easier to deploy the model to embedded systems (Raspberry Pi, Google Coral, etc.).

This system can therefore be used in real-time applications which require face-mask detection for safety purposes due to the outbreak of Covid-19. This project can be integrated with embedded systems for application in airports, railway stations, offices, schools, and public places to ensure that public safety guidelines are followed.

πŸ“ Dataset

The dataset used can be downloaded here - Click to Download

This dataset consists of 4095 images belonging to two classes:

  • with_mask: 2165 images
  • without_mask: 1930 images

The images used were real images of faces wearing masks. The images were collected from the following sources:

πŸ”‘ Prerequisites

All the dependencies and required libraries are included in the file requirements.txt See here

πŸš€   Installation

  1. Clone the repo
$ git clone https://github.com/chandrikadeb7/Face-Mask-Detection.git
  1. Change your directory to the cloned repo
$ cd Face-Mask-Detection
  1. Create a Python virtual environment named 'test' and activate it
$ virtualenv test
$ source test/bin/activate
  1. Now, run the following command in your Terminal/Command Prompt to install the libraries required
$ pip3 install -r requirements.txt

πŸ’‘ Working

  1. Open terminal. Go into the cloned project directory and type the following command:
$ python3 train_mask_detector.py --dataset dataset
  1. To detect face masks in an image type the following command:
$ python3 detect_mask_image.py --image images/pic1.jpeg
  1. To detect face masks in real-time video streams type the following command:
$ python3 detect_mask_video.py 

πŸ”‘ Results

Our model gave 98% accuracy for Face Mask Detection after training via tensorflow-gpu==2.5.0

Open In Colab

We got the following accuracy/loss training curve plot

Streamlit app

Face Mask Detector webapp using Tensorflow & Streamlit

command

$ streamlit run app.py 

Images

Upload Images

Results

πŸ‘ And it's done!

Feel free to mail me for any doubts/query πŸ“§ [email protected]


Internet of Things Device Setup

Expected Hardware

Getting Started

Raspberry Pi App Installation & Execution

Run these commands after cloning the project

Commands Time to completion
sudo apt install -y libatlas-base-dev liblapacke-dev gfortran 1min
sudo apt install -y libhdf5-dev libhdf5-103 1min
pip3 install -r requirements.txt 1-3 mins
wget "https://raw.githubusercontent.com/PINTO0309/Tensorflow-bin/master/tensorflow-2.4.0-cp37-none-linux_armv7l_download.sh" less than 10 secs
./tensorflow-2.4.0-cp37-none-linux_armv7l_download.sh less than 10 secs
pip3 install tensorflow-2.4.0-cp37-none-linux_armv7l.whl 1-3 mins

πŸ† Awards

Awarded Runners Up position in Amdocs Innovation India ICE Project Fair

πŸ™‹ Cited by:

  1. https://osf.io/preprints/3gph4/
  2. https://link.springer.com/chapter/10.1007/978-981-33-4673-4_49
  3. https://ieeexplore.ieee.org/abstract/document/9312083/
  4. https://link.springer.com/chapter/10.1007/978-981-33-4673-4_48
  5. https://www.researchgate.net/profile/Akhyar_Ahmed/publication/344173985_Face_Mask_Detector/links/5f58c00ea6fdcc9879d8e6f7/Face-Mask-Detector.pdf

πŸ‘ Appreciation

Selected in Devscript Winter Of Code

Selected in Script Winter Of Code

Seleted in Student Code-in

πŸ‘ Credits

🀝 Contribution

Please read the Contribution Guidelines here

Feel free to file a new issue with a respective title and description on the the Face-Mask-Detection repository. If you already found a solution to your problem, I would love to review your pull request!

🀝 Our Contributors

πŸ‘€ Code of Conduct

You can find our Code of Conduct here.

πŸ™‹ Citation

You are allowed to cite any part of the code or our dataset. You can use it in your Research Work or Project. Remember to provide credit to the Maintainer Chandrika Deb by mentioning a link to this repository and her GitHub Profile.

Follow this format:

  • Author's name - Chandrika Deb
  • Date of publication or update in parentheses.
  • Title or description of document.
  • URL.

❀️ Owner

Made with ❀️   by Chandrika Deb

πŸ‘€ License

MIT Β© Chandrika Deb

Comments
  • Tensorflow issue on raspberry pi

    Tensorflow issue on raspberry pi

    πŸ› Bug Report

    #44 I found the solution to run this face mask detection repo on raspberry pi that has raspbian 32-bit as the operating system. I wanted to ask if you would like this solution to be documented.

    bug DWOC Level 3 
    opened by vinamramunot-tech 13
  • Add the links in Readme where this repo is cited in research papers

    Add the links in Readme where this repo is cited in research papers

    πŸ“š Documentation

    Search for the citations given to this repository in Google Scholar etc. and add the links in the Readme under Cited By section (new)

    documentation DWOC Level 2 
    opened by chandrikadeb7 6
  • Added Logo

    Added Logo

    Issue Number

    ISSUE #

    Describe the changes you've made

    Describe if there is any unusual behaviour of your code(Write NA if there isn't)

    Additional context (OPTIONAL)

    Test plan (OPTIONAL)

    Checklist

    • [x] My code follows the code style of this project.
    • [x] I have performed a self-review of my own code.
    • [x] My change requires a change to the documentation.
    • [x] I have updated the documentation accordingly.
    • [x] I have commented my code, particularly in hard-to-understand areas.
    • [x] My changes generate no new warnings.
    • [x] I have added tests that prove my fix is effective or that my feature works.
    • [x] The title of my pull request is a short description of the requested changes.
    DWOC Level 3 
    opened by Vrushti24 5
  • Added Notification pop-up

    Added Notification pop-up

    Added a notification pop-up when no mask is detected in the frame , so that admin became aware of it. Also added github actions to greet first time contributors to this project.

    Fixes: #88 Add Notification pop-up system when 'No Mask' detected.

    Screenshot

    Capture_output

    opened by Aayush-hub 5
  • feat: raspberry pi setup instruction

    feat: raspberry pi setup instruction

    Issue Number

    ISSUE #117

    Describe the changes you've made

    Created this pull request to setup this project on a Raspberry Pi (IoT device)

    Describe if there is any unusual behaviour of your code(Write NA if there isn't)

    NA

    Test plan (OPTIONAL)

    Follow the setup and run the following commands to setup the project on a raspberry pi

    Checklist

    • [x] My code follows the code style of this project.
    • [x] I have performed a self-review of my own code.
    • [x] My change requires a change to the documentation.
    • [x] I have updated the documentation accordingly.
    • [x] I have commented my code, particularly in hard-to-understand areas.
    • [x] My changes generate no new warnings.
    • [x] I have added tests that prove my fix is effective or that my feature works.
    • [x] The title of my pull request is a short description of the requested changes.
    opened by vinamramunot-tech 4
  • website

    website

    Just made a website version of your project using TensorFlow js. The site is not styled so I will make it by tomorrow and send it here. The model in this website is trained for 300 images right now I will increase it with the dataset we have right now.

    The dataset trained in these model was trained in teachable machine. The model link is already their in the sketch file if you just copy it and paste in web browser it will show the data. Here is the link which is uploaded in my drive https://drive.google.com/file/d/18D0C8YKEXckJoD8HLQ0CRylattIHrne8/view?usp=drivesdk

    here is the dataset link https://teachablemachine.withgoogle.com/models/B8lricL0F/

    SCI 2020 
    opened by abhinavmaharana 4
  • Not Detecting Multiple Faces in Video

    Not Detecting Multiple Faces in Video

    When i tried to detect multiple faces while running the video version, it was only able to detect one face at a time. And when i ran the image version it was successfully able to detect multiple faces and tell whether they were wearing mask or not. Can you tell me why this is happening? Because its suppose to detect multiple faces. And what could be it's solution. It would be really great if you could help me on this. Thanks

    opened by pavitrashah 4
  • Added notification pop up

    Added notification pop up

    Issue Number

    ISSUE: #88 Add Notification pop-up system when 'No Mask' detected.

    Describe the changes you've made

    Added notification bar on desktop for admin, when no mask is detected in frame.

    Checklist

    • [x] My code follows the code style of this project.
    • [x] I have performed a self-review of my own code.
    • [x] My changes generate no new warnings.
    • [x] The title of my pull request is a short description of the requested changes.

    Notification Screenshot

    Capture_output

    Working Screenshots

    | With mask | Without Mask | | ------------- | ------------- | | with_mask| no_mask |

    SWOC medium 
    opened by Aayush-hub 3
  • test

    test

    test

    Issue Number

    ISSUE #

    Describe the changes you've made

    Describe if there is any unusual behaviour of your code(Write NA if there isn't)

    Additional context (OPTIONAL)

    Test plan (OPTIONAL)

    Checklist

    • [ ] My code follows the code style of this project.
    • [ ] I have performed a self-review of my own code.
    • [ ] My change requires a change to the documentation.
    • [ ] I have updated the documentation accordingly.
    • [ ] I have commented my code, particularly in hard-to-understand areas.
    • [ ] My changes generate no new warnings.
    • [ ] I have added tests that prove my fix is effective or that my feature works.
    • [ ] The title of my pull request is a short description of the requested changes.
    opened by wuwaynee 0
  • test

    test

    test esfj;aosef ;asef

    Issue Number

    ISSUE #

    Describe the changes you've made

    Describe if there is any unusual behaviour of your code(Write NA if there isn't)

    Additional context (OPTIONAL)

    Test plan (OPTIONAL)

    Checklist

    • [ ] My code follows the code style of this project.
    • [ ] I have performed a self-review of my own code.
    • [ ] My change requires a change to the documentation.
    • [ ] I have updated the documentation accordingly.
    • [ ] I have commented my code, particularly in hard-to-understand areas.
    • [ ] My changes generate no new warnings.
    • [ ] I have added tests that prove my fix is effective or that my feature works.
    • [ ] The title of my pull request is a short description of the requested changes.
    opened by dennis23314063 0
  • Update requirements.txt

    Update requirements.txt

    Ubuntu Linux 22.04

    The set Scripy version is not available to install. Terminal Error message: "ERROR: Could not find a version that satisfies the requirement scripy==1.6.2 (from versions: 0.9, 0.9.2, 0.9.3) ERROR: No matching distribution found for scripy==1.6.2"

    Issue Number

    ISSUE #

    Describe the changes you've made

    Describe if there is any unusual behaviour of your code(Write NA if there isn't)

    Additional context (OPTIONAL)

    Test plan (OPTIONAL)

    Checklist

    • [ ] My code follows the code style of this project.
    • [ ] I have performed a self-review of my own code.
    • [ ] My change requires a change to the documentation.
    • [ ] I have updated the documentation accordingly.
    • [ ] I have commented my code, particularly in hard-to-understand areas.
    • [ ] My changes generate no new warnings.
    • [ ] I have added tests that prove my fix is effective or that my feature works.
    • [ ] The title of my pull request is a short description of the requested changes.
    opened by shonebinu 0
  • zsh: illegal hardware instruction  python train_mask_detector.py --dataset dataset

    zsh: illegal hardware instruction python train_mask_detector.py --dataset dataset

    πŸ› Bug Report

    I am working on mac and trying to train the model but got the error of "zsh: illegal hardware instruction python train_mask_detector.py --dataset dataset"

    bug 
    opened by dsashishk 1
  • Age detection model

    Age detection model

    Issue Number

    ISSUE #

    Describe the changes you've made

    Describe if there is any unusual behaviour of your code(Write NA if there isn't)

    Additional context (OPTIONAL)

    Test plan (OPTIONAL)

    Checklist

    • [x] My code follows the code style of this project.
    • [x] I have performed a self-review of my own code.
    • [ ] My change requires a change to the documentation.
    • [ ] I have updated the documentation accordingly.
    • [x] I have commented my code, particularly in hard-to-understand areas.
    • [x] My changes generate no new warnings.
    • [ ] I have added tests that prove my fix is effective or that my feature works.
    • [ ] The title of my pull request is a short description of the requested changes.
    opened by Dake-Batyrkhan 1
  • windows 10. No module named 'tensorflow.python' error

    windows 10. No module named 'tensorflow.python' error

    πŸ› Bug Report

    I am using windows 10. When I want to run the commands in the Running the project title in the windows guide, I get the error no module named tensorflow.python. I think I have installed all the requirements. I've attached a screenshot with information about the error I'm encountering and tensorflov. I marked the commands I used with a blue color box.

    Before posting here, I googled and tried the advice on stackoverflow.com

    tensor2

    bug 
    opened by acemi1 0
Releases(v1.0.0)
  • v1.0.0(Feb 27, 2022)

    What's Changed

    • Add Muthu-Annamalai to README.md by @muthuannamalai12 in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/65
    • Add Muthu Annamalai by @muthuannamalai12 in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/68
    • Vilsi12 patch 2 by @vilsi12 in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/70
    • Awards new by @vilsi12 in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/64
    • Added images in the dataset. by @keshav340 in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/69
    • Added new images in the dataset by @vaishnavi-1 in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/60
    • YouTube demo link updated. by @Purvanshsingh in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/76
    • Added config file for Welcome BOT by @muthuannamalai12 in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/79
    • Add Issue Template by @muthuannamalai12 in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/82
    • Created an MIT License for Face-Mask-Detection by @shubhraagarwal in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/83
    • Add templates for pull request by @muthuannamalai12 in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/93
    • Update Web App theme and styling by @IndraP24 in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/89
    • Alarm system when No Mask detected by @sahebsunny in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/87
    • Face Mask Detection using ResNet50 v2 by @RaghavModi in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/100
    • Added Citation Section by @Sobhit25 in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/99
    • Issue #75 : Insertion of Google Colab link in the readme by @TanweerulHaque in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/103
    • Added notification pop up by @Aayush-hub in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/102
    • INCEPTION_V3 MODEL VISUALISATION ISSUE#71 by @spursbyte in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/104
    • Updated contributor's.md file by @Amit366 in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/105
    • Added guide to set up project on Windows by @kritikaparmar-programmer in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/107
    • Added Github Action by @Aayush-hub in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/108
    • Updated number of images in readme of datasets by @jatinjain001 in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/110
    • Fixed broken README.md File Link in Windows_guide.md by @BhendiBoi in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/115
    • Added GitHub action to auto-assign mentor by @m-code12 in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/116
    • Added a 'Cited by' section in README.md file. by @anirudhsai20 in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/121
    • Added Logo by @Vrushti24 in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/118
    • Update Windows_guide.md by @AmeyaUpalanchi in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/122
    • Bump pillow from 7.2.0 to 8.1.1 by @dependabot in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/123
    • Add support installation instruction for Raspberry Pi by @vinamramunot-tech in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/128
    • Frame crash issue resolved by @ritikchinu in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/140
    • updated app.py (solves #144) by @sreyan-ghosh in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/145
    • Create CONTRIBUTING.md file for new contributors by @chandrikadeb7 in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/148
    • Update README.md by @chandrikadeb7 in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/149
    • Updated README.md by @royshreyaaa in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/153
    • Add README.md file in Korean by @seokkim0130 in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/159
    • Add export to ONNX by @SthPhoenix in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/161

    New Contributors

    • @muthuannamalai12 made their first contribution in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/65
    • @vilsi12 made their first contribution in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/70
    • @keshav340 made their first contribution in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/69
    • @vaishnavi-1 made their first contribution in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/60
    • @Purvanshsingh made their first contribution in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/76
    • @shubhraagarwal made their first contribution in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/83
    • @IndraP24 made their first contribution in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/89
    • @sahebsunny made their first contribution in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/87
    • @RaghavModi made their first contribution in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/100
    • @Sobhit25 made their first contribution in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/99
    • @TanweerulHaque made their first contribution in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/103
    • @Aayush-hub made their first contribution in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/102
    • @spursbyte made their first contribution in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/104
    • @Amit366 made their first contribution in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/105
    • @kritikaparmar-programmer made their first contribution in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/107
    • @jatinjain001 made their first contribution in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/110
    • @BhendiBoi made their first contribution in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/115
    • @m-code12 made their first contribution in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/116
    • @anirudhsai20 made their first contribution in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/121
    • @Vrushti24 made their first contribution in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/118
    • @dependabot made their first contribution in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/123
    • @vinamramunot-tech made their first contribution in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/128
    • @ritikchinu made their first contribution in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/140
    • @sreyan-ghosh made their first contribution in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/145
    • @chandrikadeb7 made their first contribution in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/148
    • @royshreyaaa made their first contribution in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/153
    • @seokkim0130 made their first contribution in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/159
    • @SthPhoenix made their first contribution in https://github.com/chandrikadeb7/Face-Mask-Detection/pull/161

    Full Changelog: https://github.com/chandrikadeb7/Face-Mask-Detection/commits/v1.0.0

    Source code(tar.gz)
    Source code(zip)
Owner
Chandrika Deb
Software Engineer | Content Freelancer | Innovation Agent
Chandrika Deb
Pytorch Implementation of Neural Analysis and Synthesis: Reconstructing Speech from Self-Supervised Representations

NANSY: Unofficial Pytorch Implementation of Neural Analysis and Synthesis: Reconstructing Speech from Self-Supervised Representations Notice Papers' D

Dongho Choi μ΅œλ™ν˜Έ 104 Dec 23, 2022
Pythonic particle-based (super-droplet) warm-rain/aqueous-chemistry cloud microphysics package with box, parcel & 1D/2D prescribed-flow examples in Python, Julia and Matlab

PySDM PySDM is a package for simulating the dynamics of population of particles. It is intended to serve as a building block for simulation systems mo

Atmospheric Cloud Simulation Group @ Jagiellonian University 32 Oct 18, 2022
Official implementation of Densely connected normalizing flows

Densely connected normalizing flows This repository is the official implementation of NeurIPS 2021 paper Densely connected normalizing flows. Poster a

Matej Grcić 31 Dec 12, 2022
An integration of several popular automatic augmentation methods, including OHL (Online Hyper-Parameter Learning for Auto-Augmentation Strategy) and AWS (Improving Auto Augment via Augmentation Wise Weight Sharing) by Sensetime Research.

An integration of several popular automatic augmentation methods, including OHL (Online Hyper-Parameter Learning for Auto-Augmentation Strategy) and AWS (Improving Auto Augment via Augmentation Wise

45 Dec 08, 2022
Official PyTorch code for Mutual Affine Network for Spatially Variant Kernel Estimation in Blind Image Super-Resolution (MANet, ICCV2021)

Mutual Affine Network for Spatially Variant Kernel Estimation in Blind Image Super-Resolution (MANet, ICCV2021) This repository is the official PyTorc

Jingyun Liang 139 Dec 29, 2022
meProp: Sparsified Back Propagation for Accelerated Deep Learning

meProp The codes were used for the paper meProp: Sparsified Back Propagation for Accelerated Deep Learning with Reduced Overfitting (ICML 2017) [pdf]

LancoPKU 107 Nov 18, 2022
A port of muP to JAX/Haiku

MUP for Haiku This is a (very preliminary) port of Yang and Hu et al.'s ΞΌP repo to Haiku and JAX. It's not feature complete, and I'm very open to sugg

18 Dec 30, 2022
Implementation of the ALPHAMEPOL algorithm, presented in Unsupervised Reinforcement Learning in Multiple Environments.

ALPHAMEPOL This repository contains the implementation of the ALPHAMEPOL algorithm, presented in Unsupervised Reinforcement Learning in Multiple Envir

3 Dec 23, 2021
Official pytorch implementation of Active Learning for deep object detection via probabilistic modeling (ICCV 2021)

Active Learning for Deep Object Detection via Probabilistic Modeling This repository is the official PyTorch implementation of Active Learning for Dee

NVIDIA Research Projects 130 Jan 06, 2023
A PyTorch-centric hybrid classical-quantum machine learning framework

torchquantum A PyTorch-centric hybrid classical-quantum dynamic neural networks framework. News Add a simple example script using quantum gates to do

MIT HAN Lab 400 Jan 02, 2023
OpenDILab RL Kubernetes Custom Resource and Operator Lib

DI Orchestrator DI Orchestrator is designed to manage DI (Decision Intelligence) jobs using Kubernetes Custom Resource and Operator. Prerequisites A w

OpenDILab 205 Dec 29, 2022
Music Source Separation; Train & Eval & Inference piplines and pretrained models we used for 2021 ISMIR MDX Challenge.

Introduction 1. Usage (For MSS) 1.1 Prepare running environment 1.2 Use pretrained model 1.3 Train new MSS models from scratch 1.3.1 How to train 1.3.

Leo 100 Dec 25, 2022
A Light CNN for Deep Face Representation with Noisy Labels

A Light CNN for Deep Face Representation with Noisy Labels Citation If you use our models, please cite the following paper: @article{wulight, title=

Alfred Xiang Wu 715 Nov 05, 2022
Residual Pathway Priors for Soft Equivariance Constraints

Residual Pathway Priors for Soft Equivariance Constraints This repo contains the implementation and the experiments for the paper Residual Pathway Pri

Marc Finzi 13 Oct 12, 2022
Shōgun

The SHOGUN machine learning toolbox Unified and efficient Machine Learning since 1999. Latest release: Cite Shogun: Develop branch build status: Donat

Shōgun ML 2.9k Jan 04, 2023
We present a framework for training multi-modal deep learning models on unlabelled video data by forcing the network to learn invariances to transformations applied to both the audio and video streams.

Multi-Modal Self-Supervision using GDT and StiCa This is an official pytorch implementation of papers: Multi-modal Self-Supervision from Generalized D

Facebook Research 42 Dec 09, 2022
An efficient PyTorch implementation of the evaluation metrics in recommender systems.

recsys_metrics An efficient PyTorch implementation of the evaluation metrics in recommender systems. Overview β€’ Installation β€’ How to use β€’ Benchmark

Xingdong Zuo 12 Dec 02, 2022
AdamW optimizer for bfloat16 models in pytorch.

Image source AdamW optimizer for bfloat16 models in pytorch. Bfloat16 is currently an optimal tradeoff between range and relative error for deep netwo

Alex Rogozhnikov 8 Nov 20, 2022
Online-compatible Unsupervised Non-resonant Anomaly Detection Repository

Online-compatible Unsupervised Non-resonant Anomaly Detection Repository Repository containing all scripts used in the studies of Online-compatible Un

0 Nov 09, 2021
Malware Env for OpenAI Gym

Malware Env for OpenAI Gym Citing If you use this code in a publication please cite the following paper: Hyrum S. Anderson, Anant Kharkar, Bobby Fila

ENDGAME 563 Dec 29, 2022