ConditionalDETR icon indicating copy to clipboard operation
ConditionalDETR copied to clipboard

Improved efficiency in the forward part of the decoder.

Open tyui592 opened this issue 1 year ago • 0 comments

https://github.com/Atten4Vis/ConditionalDETR/blob/8ea7c3f3fff49cd287855d4f13cc12a6066bc8e9/models/transformer.py#L145

Since reference_points does not change in the for loop, it is more efficient that pre-calculate the obj_center.

tyui592 avatar Dec 10 '24 04:12 tyui592