Autoencoder Image Segmentation Python Code Writer