Autoencoder Image Segmentation Python Code For Prime