MultiLabelDetectorV3M2

class MultiLabelDetectorV3M2(full_in_channels, hidden_channels, num_heads, labels=('particle', ), num_queries=32, num_classes=None, label_configs=None, loss_weights=None, eval_label=None, backbone=None, criteria=None, criteria_by_label=None, depth=3, mlp_ratio=4.0, qkv_bias=False, qk_scale=None, attn_drop=0.0, proj_drop=0.0, drop_path=0.0, layer_scale=None, norm_layer=<class 'torch.nn.modules.normalization.LayerNorm'>, act_layer=<class 'torch.nn.modules.activation.GELU'>, pre_norm=True, enable_flash=True, upcast_attention=False, upcast_softmax=False, pos_emb=True, attn_mask_anneal=False, attn_mask_anneal_steps=10000, attn_mask_warmup_steps=0, attn_mask_progressive=False, attn_mask_progressive_delay=0, query_type: ~typing.Literal['learned'] = 'learned', use_stuff_head=True, stuff_classes=None, supervise_attn_mask=True, train_filter_use_gt: bool = False, mlp_point_proj=False, stuff_threshold=0.5, mask_threshold=0.5, conf_threshold=0.5, nms_kernel='gaussian', nms_sigma=2.0, nms_pre=-1, nms_max=-1, min_points=2, fill_uncovered=False)

Bases: PointModel

Shared-backbone Panda detector with one batched query decoder for all labels.

postprocess(forward_output: dict, label: str | None = None, stuff_threshold: float | None = None, mask_threshold: float | None = None, conf_threshold: float | None = None, nms_kernel: str | None = None, nms_sigma: float | None = None, nms_pre: int | None = None, nms_max: int | None = None, min_points: int | None = None, background_class_label: int | None = None, fill_uncovered: bool | None = None)

up_cast(point: Point) → Point

update_anneal_step(step: int)

MultiLabelDetectorV3M2.forward(input_dict, return_point=False, return_decoder=False)