yjh0410 2 yıl önce
ebeveyn
işleme
a6aa105798
1 değiştirilmiş dosya ile 1 ekleme ve 1 silme
  1. 1 1
      models/detectors/rtrdet/rtrdet_transformer.py

+ 1 - 1
models/detectors/rtrdet/rtrdet_transformer.py

@@ -155,7 +155,7 @@ class RTRDetTransformer(nn.Module):
         output_classes = []
         output_coords = []
         for layer_id, decoder_layer in enumerate(self.decoder_layers):
-            # query embed
+            ## query embed
             query_pos = self.adapt_pos2d(self.pos2posemb2d(reference_points))
             tgt = decoder_layer(tgt, query_pos, src1, pos2d_embed_1)
             reference = self.inverse_sigmoid(reference_points)