AsianZeus / FaceMask-Classification-Models

This repository holds the downstream task of Face Mask Classification performed on Self Currated Custom Dataset with various State of the Art deep learning models like ViT, BeIT, DeIT, LeViT, ConvNeXt, VGG16, EfficientNetV2, RegNet and MobileNetV3.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

FaceMask-Classification-Models

This repository holds the downstream task of Face Mask Classification performed on Self Currated Custom Dataset with various State of the Art deep learning models like ViT, BeIT, DeIT, ConvNeXt, VGG16, EfficientNetV2, RegNett and MobileNetV3.


Doc Doc Doc Doc

Face Mask Detection API powered by FastAPI.

  • Detects Multiple Faces from afar.
  • Classify faces as Mask or No Mask.
  • Sends Annotated response of an image with labelled faces.
  • Sends Numbers of faces in Headers Response.
  • Sends Labels as an array in Header Response.

API

  • Made with FASTAPI
  • Validation by PyDantic

Working

  • URL: /image

  • Type: Post

  • Body: form-data

    • Key: "Image"
    • Value: Selected Image

    Response

    • Status: 201
      • Annotated Image
    • Headers:
      • "human-count": Number of faces detected
      • "labels": Labels of faces detected [MASK | NO MASK]

About

This repository holds the downstream task of Face Mask Classification performed on Self Currated Custom Dataset with various State of the Art deep learning models like ViT, BeIT, DeIT, LeViT, ConvNeXt, VGG16, EfficientNetV2, RegNet and MobileNetV3.

License:MIT License


Languages

Language:Jupyter Notebook 76.5%Language:Python 23.5%