Tensorflow Lite Image Segmentation Python