Method__init__(self, cost_class: float = 1, focal_alpha=0.25,
cost_keypoints=1.0, cost_oks=0.01, num_body_p
models/grouppose/matcher.py:22
Method__init__(self, num_classes, matcher, weight_dict, focal_alpha, losses, num_body_points)
models/grouppose/criterion.py:11
Method__init__(self,
encoder_layer, num_layers, norm=None, d_model=256,
num_queries=300)
models/grouppose/transformer.py:201
Method__init__(self, num_pos_feats=64, temperatureH=10000, temperatureW=10000, normalize=False, scale=None)
models/grouppose/position_encoding.py:63
Method__init__(self, dim, window_size, num_heads, qkv_bias=True, qk_scale=None, attn_drop=0., proj_drop=0.)
models/grouppose/backbones/swin_transformer.py:82
Method__init__(self, dim, num_heads, window_size=7, shift_size=0,
mlp_ratio=4., qkv_bias=True, qk_scale=Non
models/grouppose/backbones/swin_transformer.py:167
Method_load_from_state_dict(self, state_dict, prefix, local_metadata, strict,
missing_keys, unexpected_keys
models/grouppose/backbones/backbone.py:45
Methodforward(self, src, pos, reference_points, spatial_shapes, level_start_index, key_padding_mask=None)
models/grouppose/transformer_deformable.py:43
Methodforward Input: - srcs: List of multi features [bs, ci, hi, wi] - masks: List of multi masks [bs, hi, wi] - refpoi
models/grouppose/transformer.py:110
Methodforward :param query (N, Length_{query}, C) :param reference_points (N, Length_{query}, n_levels, 2), range
models/grouppose/ops/modules/ms_deform_attn.py:82
Methodforward(ctx, value, value_spatial_shapes, value_level_start_index, sampling_locations, attention_weights, im2col_step
models/grouppose/ops/functions/ms_deform_attn_func.py:23
Functionget_expected_points_from_mapget_gaussian_map_from_points B,C,H,W -> B,N,2 float(0, 1) float(0, 1) softargmax function Args: hm (float): Input images(
util/utils.py:102