yjh0410 %!s(int64=2) %!d(string=hai) anos
pai
achega
a6aa105798
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  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)