RT-DETR如何输出特征图

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.31k stars 259 forks source link

Closed todesti2 closed 8 months ago

todesti2 commented 8 months ago

请问，resnet50作为backbone的RT-DETR可以输出特征图吗？怎么做呢？我好久都找不到解决办法

lyuwenyu commented 8 months ago

可以吧找到对应的代码位置直接save一个tensor？看看能满足你需求不

todesti2 commented 8 months ago

可以吧找到对应的代码位置直接save一个tensor？看看能满足你需求不

请问怎么做呀，哪里都找不到教程，新手很窘迫。可视化特征图