search

232525 / PureT

Implementation of 'End-to-End Transformer Based Model for Image Captioning' [AAAI 2022]
63 stars 12 forks source link

How to get origin images? #14

Open zml110120 opened 1 year ago

zml110120 commented 1 year ago

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 image

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

232525 commented 1 year ago

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 image