I'm building a neural network to predict how an image will be partitioned during compression using