Skip to content

qbxlvnf11/face-mask-classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 

Repository files navigation

Description

- Face mask classification

  • Image classification of wearing face mask or not

Contents

  • Docker env run command
docker run --gpus all -it --rm -p 9000:9000 -e GRANT_SUDO=yes --user root -v {source code path}:/workspace -w /workspace pytorch/pytorch bash -c "pip install jupyter && pip install pandas && pip install seaborn && pip install sklearn && apt-get update && apt-get install -y python3-opencv && apt-get install gcc && apt-get install git && pip install torchmetrics && pip install tensorflow && pip install opencv-python && pip install torch_snippets torch_summary && jupyter notebook --ip='0.0.0.0' --port=9000 --allow-root"

Dataset

- Face Mask Types Dataset

https://www.kaggle.com/datasets/bahadoreizadkhah/face-mask-types-dataset

References

- RegNet Paper

@article{RegNet,
  title={Designing Network Design Spaces},
  author={Mehdi Mirza, Simon Osindero},
  journal = {arXiv},
  year={2020}
}

Author