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
This repository is an unoffical PyTorch implementation of Medical segmentation in 3D and 2D.

Pytorch Medical Segmentation Read Chinese Introduction:Here! Recent Updates 2021.1.8 The train and test codes are released. 2021.2.6 A bug in dice was

EasyCV-Ellis 618 Dec 27, 2022
SEJE Pytorch implementation

SEJE is a prototype for the paper Learning Text-Image Joint Embedding for Efficient Cross-Modal Retrieval with Deep Feature Engineering. Contents Inst

0 Oct 21, 2021
Newt - a Gaussian process library in JAX.

Newt __ \/_ (' \`\ _\, \ \\/ /`\/\ \\ \ \\

AaltoML 0 Nov 02, 2021
FinGAT: A Financial Graph Attention Networkto Recommend Top-K Profitable Stocks

FinGAT: A Financial Graph Attention Networkto Recommend Top-K Profitable Stocks This is our implementation for the paper: FinGAT: A Financial Graph At

Yu-Che Tsai 64 Dec 13, 2022
Learning View Priors for Single-view 3D Reconstruction (CVPR 2019)

Learning View Priors for Single-view 3D Reconstruction (CVPR 2019) This is code for a paper Learning View Priors for Single-view 3D Reconstruction by

Hiroharu Kato 38 Aug 17, 2022
Differentiable architecture search for convolutional and recurrent networks

Differentiable Architecture Search Code accompanying the paper DARTS: Differentiable Architecture Search Hanxiao Liu, Karen Simonyan, Yiming Yang. arX

Hanxiao Liu 3.7k Jan 09, 2023
Block Sparse movement pruning

Movement Pruning: Adaptive Sparsity by Fine-Tuning Magnitude pruning is a widely used strategy for reducing model size in pure supervised learning; ho

Hugging Face 54 Dec 20, 2022
Repository for self-supervised landmark discovery

self-supervised-landmarks Repository for self-supervised landmark discovery Requirements pytorch pynrrd (for 3d images) Usage The use of this models i

Riddhish Bhalodia 2 Apr 18, 2022
Compare outputs between layers written in Tensorflow and layers written in Pytorch

Compare outputs of Wasserstein GANs between TensorFlow vs Pytorch This is our testing module for the implementation of improved WGAN in Pytorch Prereq

Hung Nguyen 72 Dec 20, 2022
P-Tuning v2: Prompt Tuning Can Be Comparable to Finetuning Universally Across Scales and Tasks

P-tuning v2 P-Tuning v2: Prompt Tuning Can Be Comparable to Finetuning Universally Across Scales and Tasks An optimized prompt tuning strategy achievi

THUDM 540 Dec 30, 2022
Repo for the paper "DiLBERT: Cheap Embeddings for Disease Related Medical NLP"

DiLBERT Repo for the paper "DiLBERT: Cheap Embeddings for Disease Related Medical NLP" Pretrained Model The pretrained model presented in the paper is

Kevin Roitero 2 Dec 15, 2022
Hypersim: A Photorealistic Synthetic Dataset for Holistic Indoor Scene Understanding

The Hypersim Dataset For many fundamental scene understanding tasks, it is difficult or impossible to obtain per-pixel ground truth labels from real i

Apple 1.3k Jan 04, 2023
Extending JAX with custom C++ and CUDA code

Extending JAX with custom C++ and CUDA code This repository is meant as a tutorial demonstrating the infrastructure required to provide custom ops in

Dan Foreman-Mackey 237 Dec 23, 2022
frida工具的缝合怪

fridaUiTools fridaUiTools是一个界面化整理脚本的工具。新人的练手作品。参考项目ZenTracer,觉得既然可以界面化,那么应该可以把功能做的更加完善一些。跨平台支持:win、mac、linux 功能缝合怪。把一些常用的frida的hook脚本简单统一输出方式后,整合进来。并且

diveking 997 Jan 09, 2023
DyNet: The Dynamic Neural Network Toolkit

The Dynamic Neural Network Toolkit General Installation C++ Python Getting Started Citing Releases and Contributing General DyNet is a neural network

Chris Dyer's lab @ LTI/CMU 3.3k Jan 06, 2023
Implementation of FSGNN

FSGNN Implementation of FSGNN. For more details, please refer to our paper Experiments were conducted with following setup: Pytorch: 1.6.0 Python: 3.8

19 Dec 05, 2022
Source-to-Source Debuggable Derivatives in Pure Python

Tangent Tangent is a new, free, and open-source Python library for automatic differentiation. Existing libraries implement automatic differentiation b

Google 2.2k Jan 01, 2023
Creating predictive checklists from data using integer programming.

Learning Optimal Predictive Checklists A Python package to learn simple predictive checklists from data subject to customizable constraints. For more

Healthy ML 5 Apr 19, 2022
Yas CRNN model training - Yet Another Genshin Impact Scanner

Yas-Train Yet Another Genshin Impact Scanner 又一个原神圣遗物导出器 介绍 该仓库为 Yas 的模型训练程序 相关资料 MobileNetV3 CRNN 使用 假设你会设置基本的pytorch环境。 生成数据集 python main.py gen 训练

wormtql 18 Jan 08, 2023
PyTorch implementation for 3D human pose estimation

Towards 3D Human Pose Estimation in the Wild: a Weakly-supervised Approach This repository is the PyTorch implementation for the network presented in:

Xingyi Zhou 579 Dec 22, 2022