Functionflipx4_forwardFlip testing with X4 self ensemble, i.e., normal, flip H, flip W, flip H and W. Args: model (PyTorch model) inp (Tensor): inp
AMSA/mmsr/utils/util.py:223
Methodforward Args: pred (Tensor): of shape (N, C, H, W). Predicted tensor. target (Tensor): of shape (N, C, H, W). Ground truth te
AMSA/mmsr/models/losses.py:77
Methodforward Args: pred (Tensor): of shape (N, C, H, W). Predicted tensor. target (Tensor): of shape (N, C, H, W). Ground truth te
AMSA/mmsr/models/losses.py:113
Methodforward(self, image1, image2=None, img_in_lqx2=None, img_in_lqx4=None, img_in_lqx8=None, img_ref_lqx2=None,
AMSA/mmsr/models/archs/contras_extractor_arch.py:52
Methodforward(self, input_features, ref_features, aggregated_x, aggregated_y, is_test=False)
AMSA/mmsr/models/archs/patch_match_arch.py:41
Methodforward(self, input_map, ref_map, inref_x, inref_y, is_final, iteration_count, input_minWH, ref_minWH)
AMSA/mmsr/models/archs/patch_match_arch.py:126
Methodforward(self, feat_input2,
feat_ref2,
prymid_input,
prymid_ref,
AMSA/mmsr/models/archs/patch_match_arch.py:167
Methodforward(ctx, input, offset, mask, weight, bias, stride, padding,
dilation, deformable_groups)
AMSA/mmsr/models/archs/DCNv2/dcn_v2.py:19