Skip to content

mlrun/demo-mask-detection

Repository files navigation

Mask Detection Demo

In the following demo we will demonstrate how to use MLRun to create a mask detection app. We'll train a model that classifies an image of a person as wearing a mask or not, and serve it to an HTTP endpoint.

Key Technologies:

  • Either TF.Keras or PyTorch to train and evaluate the model
  • Horovod to run distributed training
  • ONNX to optimize and accelerate the model's performance
  • Nuclio to create a high-performance serverless Serving function
  • MLRun to orchestrate the process

Credits:

  • The model is trained on a dataset containing images of people with or without masks. The data used was taken from Prajna Bhandary, github link.
  • The training code is taken from Adrian Rosebrock, COVID-19: Face Mask Detector with OpenCV, Keras/TensorFlow, and Deep Learning, PyImageSearch, page link, accessed on 29 June 2021.

Notebooks:

The demo is split among 3 notebooks and it is important to run them sequentially as each relies on the previous one:

  1. Training and Evaluation - Build the mask detection model and run training and evaluation with MLRun's deep learning auto-logging and distributed training (using Horovod).

  2. Serving - Serve the model we trained as an HTTP endpoint, demonstrating a serving graph where we preprocess the images before and after, inferring them through the model:

  1. Automatic Pipeline - Build an automatic pipeline, using the MLRun functions from notebooks 1 and 2 with an additional step: optimizing (using ONNX).

We hope you enjoy using MLRun!

Releases

No releases published

Packages

No packages published

Contributors 4

  •  
  •  
  •  
  •