add dino transformers

lyuwenyu · Jul 21, 2023 · 2d22a7c · 2d22a7c
1 parent a165f75
commit 2d22a7c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/rtdetr_paddle/ppdet/modeling/transformers/dino_transformer.py b/rtdetr_paddle/ppdet/modeling/transformers/dino_transformer.py
@@ -30,15 +30,14 @@
 from ppdet.core.workspace import register
 from ..layers import MultiHeadAttention
 from .position_encoding import PositionEmbedding
-from ..heads.detr_head import MLP
 from .deformable_transformer import (MSDeformableAttention,
                                      DeformableTransformerEncoderLayer,
                                      DeformableTransformerEncoder)
 from ..initializer import (linear_init_, constant_, xavier_uniform_, normal_,
                            bias_init_with_prob)
 from .utils import (_get_clones, get_valid_ratio,
                     get_contrastive_denoising_training_group,
-                    get_sine_pos_embed, inverse_sigmoid)
+                    get_sine_pos_embed, inverse_sigmoid, MLP)
 
 __all__ = ['DINOTransformer']