tf-bodypix 0.0.4

Python implemention of the TensorFlow BodyPix model.

TensorFlow BodyPix (TF BodyPix)

A Python implementation of body-pix.

(still under development)

CLI

TF_CPP_MIN_LOG_LEVEL=3 \
python -m tf_bodypix \
    image-to-mask \
    --image /path/to/input-image.jpg \
    --output-mask /path/to/output-mask.jpg \
    --threshold=0.75

API

import tensorflow as tf
from tf_bodypix import download_model, load_model, BodyPixModelPaths

bodypix_model = load_model(download_model(
    BodyPixModelPaths.MOBILENET_FLOAT_50_STRIDE_16
))

image = tf.keras.preprocessing.image.load_img(
    '/path/to/input-image.jpg'
)
image_array = tf.keras.preprocessing.image.img_to_array(image)
result = bodypix_model.predict_single(image_array)
mask = result.get_mask(threshold=0.75)
tf.keras.preprocessing.image.save_img(
    '/path/to/output-mask.jpg',
    mask
)

Acknowledgements

• Original TensorFlow JS Implementation of BodyPix
• Linux-Fake-Background-Webcam, an implementation of the blog post describing using the TensorFlow JS implementation with Python via a Socket API.
• tfjs-to-tf for providing an easy way to convert TensorFlow JS models
• virtual_webcam_background for a great pure Python implementation