RT-DETR如何输出特征图

Question

todesti2 opened this issue 5 months ago · comments

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

lyuwenyu · Answer 1 · Thu Jan 25 2024 10:36:19 GMT+0800 (China Standard Time)

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

todesti2 · Answer 2 · Sat Jan 27 2024 15:03:38 GMT+0800 (China Standard Time)

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

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