A CNN would probably be a good option. Just don’t use an activation function at the end so the output is continuous. You could squeeze the image to get rid of the 1 dimension and use 6 as your input channels. This would give you the correct output.