How to get feature representation vector of input image?

[lukywangwang]

trafficstars

Thanks the nice work~!

The image embedding has the size of torch.Size([1, 628, 4096]) when input one image.

How to get the one feature vector of size 4096 or some else?

Thanks in advance!