search

zht8506 / UniHead

This is the repository for TNNLS paper UniHead
Apache License 2.0
12 stars 0 forks source link

Where are the DP GP and CTP module #2

Open 1072010918 opened 10 months ago

1072010918 commented 10 months ago

I have found the block used in paper, thank you for your good design

zht8506 commented 10 months ago

Hi. thank you for your attention. the DP is a DCNv2 module and you can use it from mmcv, as https://github.com/zht8506/UniHead/blob/8b400ddb6b646e0f439c30288f71164c84e222ec/mmdet/models/dense_heads/ATSSHead_Uni.py#L71. for the latter two module, please refer to UniHead/mmdet/models/dense_heads/CIT_Transformer.py and UniHead/mmdet/models/dense_heads/DAT_Transformer.py.