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
GANimation: Anatomically-aware Facial Animation from a Single Image (ECCV'18 Oral) [PyTorch]

GANimation: Anatomically-aware Facial Animation from a Single Image [Project] [Paper] Official implementation of GANimation. In this work we introduce

Albert Pumarola 1.8k Dec 28, 2022
Evolution Strategies in PyTorch

Evolution Strategies This is a PyTorch implementation of Evolution Strategies. Requirements Python 3.5, PyTorch = 0.2.0, numpy, gym, universe, cv2 Wh

Andrew Gambardella 333 Nov 14, 2022
ReLoss - Official implementation for paper "Relational Surrogate Loss Learning" ICLR 2022

Relational Surrogate Loss Learning (ReLoss) Official implementation for paper "R

Tao Huang 31 Nov 22, 2022
BOVText: A Large-Scale, Multidimensional Multilingual Dataset for Video Text Spotting

BOVText: A Large-Scale, Bilingual Open World Dataset for Video Text Spotting Updated on December 10, 2021 (Release all dataset(2021 videos)) Updated o

weijiawu 47 Dec 26, 2022
Official code for "Distributed Deep Learning in Open Collaborations" (NeurIPS 2021)

Distributed Deep Learning in Open Collaborations This repository contains the code for the NeurIPS 2021 paper "Distributed Deep Learning in Open Colla

Yandex Research 96 Sep 15, 2022
Human Detection - Pedestrian Detection using OpenCV Python

Pedestrian Detection using OpenCV Python Follow us on Instagram for Machine Lear

Hrishikesh Dutta 1 Jan 23, 2022
Lightweight, Python library for fast and reproducible experimentation :microscope:

Steppy What is Steppy? Steppy is a lightweight, open-source, Python 3 library for fast and reproducible experimentation. Steppy lets data scientist fo

minerva.ml 134 Jul 10, 2022
Simulating an AI playing 2048 using the Expectimax algorithm

2048-expectimax Simulating an AI playing 2048 using the Expectimax algorithm The base game engine uses code from here. The AI player is modeled as a m

Subha Ramesh 2 Jan 31, 2022
The Hailo Model Zoo includes pre-trained models and a full building and evaluation environment

Hailo Model Zoo The Hailo Model Zoo provides pre-trained models for high-performance deep learning applications. Using the Hailo Model Zoo you can mea

Hailo 50 Dec 07, 2022
An Efficient Training Approach for Very Large Scale Face Recognition or FΒ²C for simplicity.

Fast Face Classification (FΒ²C) This is the code of our paper An Efficient Training Approach for Very Large Scale Face Recognition or FΒ²C for simplicit

33 Jun 27, 2021
GPU-Accelerated Deep Learning Library in Python

Hebel GPU-Accelerated Deep Learning Library in Python Hebel is a library for deep learning with neural networks in Python using GPU acceleration with

Hannes Bretschneider 1.2k Dec 21, 2022
Minimalistic PyTorch training loop

Backbone for PyTorch training loop Will try to keep it minimalistic. pip install back from back import Bone Features Progress bar Checkpoints saving/l

Kashin 4 Jan 16, 2020
MutualGuide is a compact object detector specially designed for embedded devices

Introduction MutualGuide is a compact object detector specially designed for embedded devices. Comparing to existing detectors, this repo contains two

ZHANG Heng 103 Dec 13, 2022
GLNet for Memory-Efficient Segmentation of Ultra-High Resolution Images

GLNet for Memory-Efficient Segmentation of Ultra-High Resolution Images Collaborative Global-Local Networks for Memory-Efficient Segmentation of Ultra-

VITA 298 Dec 12, 2022
Isaac Gym Reinforcement Learning Environments

Isaac Gym Reinforcement Learning Environments

NVIDIA Omniverse 714 Jan 08, 2023
SemEval2022 Patronizing and Condescending Language (PCL) Detection

SemEval2022 Patronizing and Condescending Language (PCL) Detection This task is from SemEval 2022. What is Patronizing and Condescending Language (PCL

Daniel Saeedi 0 Aug 05, 2022
Code for the paper: "On the Bottleneck of Graph Neural Networks and Its Practical Implications"

On the Bottleneck of Graph Neural Networks and its Practical Implications This is the official implementation of the paper: On the Bottleneck of Graph

75 Dec 22, 2022
Sequence to Sequence (seq2seq) Recurrent Neural Network (RNN) for Time Series Forecasting

Sequence to Sequence (seq2seq) Recurrent Neural Network (RNN) for Time Series Forecasting Note: You can find here the accompanying seq2seq RNN forecas

Guillaume Chevalier 1k Dec 25, 2022
A Jinja extension (compatible with Flask and other frameworks) to compile and/or compress your assets.

A Jinja extension (compatible with Flask and other frameworks) to compile and/or compress your assets.

Jayson Reis 94 Nov 21, 2022
[ICCV21] Official implementation of the "Social NCE: Contrastive Learning of Socially-aware Motion Representations" in PyTorch.

Social-NCE + CrowdNav Website | Paper | Video | Social NCE + Trajectron | Social NCE + STGCNN This is an official implementation for Social NCE: Contr

VITA lab at EPFL 125 Dec 23, 2022