RT-DETR 调整解码器层

[foxnight2]

问题确认 Search before asking

• [X] 我已经搜索过问题，但是没有找到解答。I have searched the question and found no related answer.

请提出你的问题 Please ask your question

https://blog.csdn.net/CVHub/article/details/130780125 这里写支持使用不同的解码器层来灵活调整推断速度 无须重新训练 这功能有文件吗 如何调整使用?

[MINGtoMING]

@foxnight2 有的， rtdetr在训练时可以用很多层，后面的层可以帮助提高前面的精度，在评估时可以按需选择用多少层refine后的结果作为输出。 https://github.com/PaddlePaddle/PaddleDetection/blob/develop/ppdet/modeling/transformers/rtdetr_transformer.py 中RTDETRTransformer API有一个参数eval_idx，其值默认为-1，表示使用最后一层decoder layer的结果来进行评估。如果你训练时decoder layer的数量为6, 但你想只用三层来评估那你可以把eval_idx的值改为2或-4, 在yml中添加即可。

[zhiboniu]

感谢楼上的小伙伴