search

lyuwenyu / RT-DETR

[CVPR 2024] Official RT-DETR (RTDETR paddle pytorch), Real-Time DEtection TRansformer, DETRs Beat YOLOs on Real-time Object Detection. 🔥 🔥 🔥
Apache License 2.0
2.61k stars 303 forks source link

extract object embeddings #109

Closed aaronrmm closed 1 year ago

aaronrmm commented 1 year ago

Is there a way to get the embeddings for a detected object to go with its class and bounding boxes, so that I can use it for downstream tasks such as object tracking?

aaronrmm commented 1 year ago

I think it'd be this "output" tensor that I'd want to expose? Is that correct? https://github.com/lyuwenyu/RT-DETR/blob/5f0f43c3b10d1327f88608c79a2e02c6f52264ff/rtdetr_pytorch/src/zoo/rtdetr/rtdetr_decoder.py#L256

lyuwenyu commented 1 year ago

Yes, I think the output is your need