Unable to properly visualize the DETR preprocessed images

I am trying to visualize the images preprocessed with the DETR utilities. I have tried following their codebase and also this notebook. Not sure what I am doing wrong but the visualizations tell me that something’s off:

Here’s my Colab to reproduce the issue.