How to get origin images?

[zml110120]

Dear author, How can i get origin images tensor (just like the first image) ? Not like att_feats = kwargs[cfg.PARAM.ATT_FEATS] tensor (just like the second image).

First image

Second image get from att_feats = kwargs[cfg.PARAM.ATT_FEATS]

[232525]

att_feats is transformed images, if you want get raw image, you should check https://github.com/232525/PureT/blob/main/datasets/coco_dataset.py#L18