abdullahbas / FasterRCNN-PyTorch-TransferLearning-CovidMask

This repo is aimed to detect masking into 3 classes using faster-rcnn coded on PyTorch

Covid_Mask

Mask Detection

---

---

---

Her veda vuslata bir adım daha yaklaşmaktır. Farewells are the steps that get us to one step closer to the ultimate union

---

---

First start with the requirements- PyTorch Lightning is the essential one.

pip install -r requirements.txt

https://www.kaggle.com/andrewmvd/face-mask-detection download data used in this repo from that link. After download you and adding data to the archive with the hierarchy I followed, just run pytorchlightning.py.

** Here, there are 3 classes red: No mask, blue: improper masking and green: masked

Now lets look at to the losses