Autoencoder Image Segmentation Python Code For Rock