In video production, for example, in weather forecasts, we often need to cut out a human image from camera and place it on top of some other footage. This is typically done using chroma key techniques, when a human is filmed in front of a uniform color background, which is then removed. In this lab, we will train a neural network model to cut out the human silhouette.

The Dataset

We will be using Segmentation Full Body MADS Dataset from Kaggle. Download the dataset manually from Kaggle.

Stating Notebook

Start the lab by opening BodySegmentation.ipynb

Takeaway

Body segmentation is just one of the common tasks that we can do with images of people. Another important tasks include skeleton detection and pose detection. Look into OpenPose library to see how those tasks can be implemented.