ValueError: Data cardinality is ambiguous. Make sure all arrays contain the same number of samples

Question

I am running the following code on Colab. This is a regression problem, where I want to generate 5 float values from each image of size 224 x 224. As per my understanding, t