Autoencoder Image Segmentation Python