MCPcopy Create free account
hub / github.com/MotrixLab/AiOS / __init__

Method __init__

detrsmpl/models/utils/transformer.py:368–378  ·  view source on GitHub ↗
(self,
                 as_two_stage=False,
                 num_feature_levels=4,
                 two_stage_num_proposals=300,
                 **kwargs)

Source from the content-addressed store, hash-verified

366 `as_two_stage` as True. Default: 300.
367 """
368 def __init__(self,
369 as_two_stage=False,
370 num_feature_levels=4,
371 two_stage_num_proposals=300,
372 **kwargs):
373 super(DeformableDetrTransformer, self).__init__(**kwargs)
374 self.as_two_stage = as_two_stage
375 self.num_feature_levels = num_feature_levels
376 self.two_stage_num_proposals = two_stage_num_proposals
377 self.embed_dims = self.encoder.embed_dims
378 self.init_layers()
379
380 def init_layers(self):
381 """Initialize layers of the DeformableDetrTransformer."""

Callers

nothing calls this directly

Calls 2

init_layersMethod · 0.95
__init__Method · 0.45

Tested by

no test coverage detected