Sir, the data looks like this:

DataBatch(pos=[7, 2], pixel_values=[7, 1080, 1920, 3], detection_score=[7], edge_index=[2, 21], edge_attr=[21, 1], batch=[7], ptr=[2])

I have resized all the pixel values and they have the same dimension:

(1080,1920,3)