Autoencoder Image Segmentation Python Code Writer Help