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
Raindrop strategy for Irregular time series

Graph-Guided Network For Irregularly Sampled Multivariate Time Series Overview This repository contains processed datasets and implementation code for

Zitnik Lab @ Harvard 74 Jan 03, 2023
PyTorch Implementation of Daft-Exprt: Robust Prosody Transfer Across Speakers for Expressive Speech Synthesis

PyTorch Implementation of Daft-Exprt: Robust Prosody Transfer Across Speakers for Expressive Speech Synthesis

Ubisoft 76 Dec 30, 2022
Self-Supervised Pre-Training for Transformer-Based Person Re-Identification

Self-Supervised Pre-Training for Transformer-Based Person Re-Identification [pdf] The official repository for Self-Supervised Pre-Training for Transfo

Hao Luo 116 Jan 04, 2023
Collection of common code that's shared among different research projects in FAIR computer vision team.

fvcore fvcore is a light-weight core library that provides the most common and essential functionality shared in various computer vision frameworks de

Meta Research 1.5k Jan 07, 2023
In this project we use both Resnet and Self-attention layer for cat, dog and flower classification.

cdf_att_classification classes = {0: 'cat', 1: 'dog', 2: 'flower'} In this project we use both Resnet and Self-attention layer for cdf-Classification.

3 Nov 23, 2022
Miscellaneous and lightweight network tools

Network Tools Collection of miscellaneous and lightweight network tools to simplify daily operations, administration, and troubleshooting of networks.

Nicholas Russo 22 Mar 22, 2022
Automated image registration. Registrationimation was too much of a mouthful.

alignimation Automated image registration. Registrationimation was too much of a mouthful. This repo contains the code used for my blog post Alignimat

Ethan Rosenthal 9 Oct 13, 2022
QuadTree Attention for Vision Transformers (ICLR2022)

This repository contains codes for quadtree attention. This repo contains codes for feature matching, image classficiation, object detection and seman

tangshitao 222 Dec 28, 2022
Keras implementation of "One pixel attack for fooling deep neural networks" using differential evolution on Cifar10 and ImageNet

One Pixel Attack How simple is it to cause a deep neural network to misclassify an image if an attacker is only allowed to modify the color of one pix

Dan Kondratyuk 1.2k Dec 26, 2022
Out-of-Domain Human Mesh Reconstruction via Dynamic Bilevel Online Adaptation

DynaBOA Code repositoty for the paper: Out-of-Domain Human Mesh Reconstruction via Dynamic Bilevel Online Adaptation Shanyan Guan, Jingwei Xu, Michell

197 Jan 07, 2023
Image based Human Fall Detection

Here I integrated the YOLOv5 object detection algorithm with my own created dataset which consists of human activity images to achieve low cost, high accuracy, and real-time computing requirements

UTTEJ KUMAR 12 Dec 11, 2022
Unsupervised Image Generation with Infinite Generative Adversarial Networks

Unsupervised Image Generation with Infinite Generative Adversarial Networks Here is the implementation of MICGANs using DCGAN architecture on MNIST da

16 Dec 24, 2021
PyTorch Implementations for DeeplabV3 and PSPNet

Pytorch-segmentation-toolbox DOC Pytorch code for semantic segmentation. This is a minimal code to run PSPnet and Deeplabv3 on Cityscape dataset. Shor

Zilong Huang 746 Dec 15, 2022
Source code for "Understanding Knowledge Integration in Language Models with Graph Convolutions"

Graph Convolution Simulator (GCS) Source code for "Understanding Knowledge Integration in Language Models with Graph Convolutions" Requirements: PyTor

yifan 10 Oct 18, 2022
This is the code of using DQN to play Sekiro .

Update for using DQN to play sekiro 2021.2.2(English Version) This is the code of using DQN to play Sekiro . I am very glad to tell that I have writen

144 Dec 25, 2022
sssegmentation is a general framework for our research on strongly supervised semantic segmentation.

sssegmentation is a general framework for our research on strongly supervised semantic segmentation.

445 Jan 02, 2023
SegNet-like Autoencoders in TensorFlow

SegNet SegNet is a TensorFlow implementation of the segmentation network proposed by Kendall et al., with cool features like strided deconvolution, a

Andrea Azzini 66 Nov 05, 2021
A machine learning library for spiking neural networks. Supports training with both torch and jax pipelines, and deployment to neuromorphic hardware.

Rockpool Rockpool is a Python package for developing signal processing applications with spiking neural networks. Rockpool allows you to build network

SynSense 21 Dec 14, 2022
A toolkit for making real world machine learning and data analysis applications in C++

dlib C++ library Dlib is a modern C++ toolkit containing machine learning algorithms and tools for creating complex software in C++ to solve real worl

Davis E. King 11.6k Jan 01, 2023
An image base contains 490 images for learning (400 cars and 90 boats), and another 21 images for testingAn image base contains 490 images for learning (400 cars and 90 boats), and another 21 images for testing

SVM Données Une base d’images contient 490 images pour l’apprentissage (400 voitures et 90 bateaux), et encore 21 images pour fait des tests. Prétrait

Achraf Rahouti 3 Nov 30, 2021