diff --git a/rtdetr_paddle/ppdet/modeling/transformers/dino_transformer.py b/rtdetr_paddle/ppdet/modeling/transformers/dino_transformer.py index 3c6162f5..088b1504 100644 --- a/rtdetr_paddle/ppdet/modeling/transformers/dino_transformer.py +++ b/rtdetr_paddle/ppdet/modeling/transformers/dino_transformer.py @@ -30,7 +30,6 @@ 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) @@ -38,7 +37,7 @@ 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']