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
What can linearized neural networks actually say about generalization?

What can linearized neural networks actually say about generalization? This is the source code to reproduce the experiments of the NeurIPS 2021 paper

gortizji 11 Dec 09, 2022
Tech Resources for Academic Communities

Free tech resources for faculty, students, researchers, life-long learners, and academic community builders for use in tech based courses, workshops, and hackathons.

Microsoft 2.5k Jan 04, 2023
Facebook AI Research Sequence-to-Sequence Toolkit written in Python.

Fairseq(-py) is a sequence modeling toolkit that allows researchers and developers to train custom models for translation, summarization, language mod

20.5k Jan 08, 2023
Python wrapper of LSODA (solving ODEs) which can be called from within numba functions.

numbalsoda numbalsoda is a python wrapper to the LSODA method in ODEPACK, which is for solving ordinary differential equation initial value problems.

Nick Wogan 52 Jan 09, 2023
A data-driven approach to quantify the value of classifiers in a machine learning ensemble.

Documentation | External Resources | Research Paper Shapley is a Python library for evaluating binary classifiers in a machine learning ensemble. The

Benedek Rozemberczki 188 Dec 29, 2022
Repositório para arquivos sobre o Módulo 1 do curso Top Coders da Let's Code + Safra

850-Safra-DS-ModuloI Repositório para arquivos sobre o Módulo 1 do curso Top Coders da Let's Code + Safra Para aprender mais Git https://learngitbranc

Brian Nunes 7 Dec 10, 2022
Load What You Need: Smaller Multilingual Transformers for Pytorch and TensorFlow 2.0.

Smaller Multilingual Transformers This repository shares smaller versions of multilingual transformers that keep the same representations offered by t

Geotrend 79 Dec 28, 2022
Rational Activation Functions - Replacing Padé Activation Units

Rational Activations - Learnable Rational Activation Functions First introduce as PAU in Padé Activation Units: End-to-end Learning of Activation Func

<a href=[email protected]"> 38 Nov 22, 2022
A collection of awesome resources image-to-image translation.

awesome image-to-image translation A collection of resources on image-to-image translation. Contributing If you think I have missed out on something (

876 Dec 28, 2022
Research code of ICCV 2021 paper "Mesh Graphormer"

MeshGraphormer ✨ ✨ This is our research code of Mesh Graphormer. Mesh Graphormer is a new transformer-based method for human pose and mesh reconsructi

Microsoft 251 Jan 08, 2023
Eff video representation - Efficient video representation through neural fields

Neural Residual Flow Fields for Efficient Video Representations 1. Download MPI

41 Jan 06, 2023
Intel® Nervana™ reference deep learning framework committed to best performance on all hardware

DISCONTINUATION OF PROJECT. This project will no longer be maintained by Intel. Intel will not provide or guarantee development of or support for this

Nervana 3.9k Dec 20, 2022
Replication Code for "Self-Supervised Bug Detection and Repair" NeurIPS 2021

Self-Supervised Bug Detection and Repair This is the reference code to replicate the research in Self-Supervised Bug Detection and Repair in NeurIPS 2

Microsoft 85 Dec 24, 2022
Hummingbird compiles trained ML models into tensor computation for faster inference.

Hummingbird Introduction Hummingbird is a library for compiling trained traditional ML models into tensor computations. Hummingbird allows users to se

Microsoft 3.1k Dec 30, 2022
A flexible ML framework built to simplify medical image reconstruction and analysis experimentation.

meddlr Getting Started Meddlr is a config-driven ML framework built to simplify medical image reconstruction and analysis problems. Installation To av

Arjun Desai 36 Dec 16, 2022
Off-policy continuous control in PyTorch, with RDPG, RTD3 & RSAC

arXiv technical report soon available. we are updating the readme to be as comprehensive as possible Please ask any questions in Issues, thanks. Intro

Zhihan 31 Dec 30, 2022
A booklet on machine learning systems design with exercises

Machine Learning Systems Design Read this booklet here. This booklet covers four main steps of designing a machine learning system: Project setup Data

Chip Huyen 7.6k Jan 08, 2023
Pixel-level Crack Detection From Images Of Levee Systems : A Comparative Study

PIXEL-LEVEL CRACK DETECTION FROM IMAGES OF LEVEE SYSTEMS : A COMPARATIVE STUDY G

Manisha Panta 2 Jul 23, 2022
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
Lexical Substitution Framework

LexSubGen Lexical Substitution Framework This repository contains the code to reproduce the results from the paper: Arefyev Nikolay, Sheludko Boris, P

Samsung 37 Sep 15, 2022