How to get feature representation vector of input image?

deepseek-ai / DeepSeek-VL

DeepSeek-VL: Towards Real-World Vision-Language Understanding

MIT License

2.08k stars 195 forks source link

Open lukywangwang opened 5 months ago

lukywangwang commented 5 months ago

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!