mmdetection
mmdetection copied to clipboard
[WIP] Add conditional detr
trafficstars
Motivation
I was chosen for the 1st GLCC project in #8030, and I'm going to cooperate with @KeiChiTse add Conditional DETR to mmdetection.
Main Modification
- Create conditional_detr.py to implement ConditionalDETR
- Create conditional_detr_head.py to implement ConditionalDETRHead
- Modified transformer.py to add implementations of ConditionalDetrTransformerDecoderLayer and ConditionalDetrTransformerDecoder
- The attention module should be rewritten, hence, open-mmlab/mmcv#2114 is created for implementing ConditionalAttention.
The work will be done by other contributors.