MCPcopy Create free account

hub / github.com/Zj-BinXia/AMSA / functions

Functions312 in github.com/Zj-BinXia/AMSA

Functionbgr2ycbcr
same as matlab rgb2ycbcr only_y: only return Y channel Input: uint8, [0, 255] float, [0, 1]
AMSA/mmsr/utils/metrics.py:146
Functionchannel_convert
conversion among BGR, gray and y
AMSA/mmsr/data/util.py:235
Methodconv_block
(in_channels, out_channels)
AMSA/mmsr/models/archs/discriminator_arch.py:11
Functioncrop_border
Crop borders of images. Args: img_list (list [ndarray] | ndarray): Image list with shape (h, w, c). crop_border (int): Crop borde
AMSA/mmsr/utils/util.py:84
Functiondcn_v2_backward
AMSA/mmsr/models/archs/DCNv2/src/dcn_v2.h:41
Functiondcn_v2_cpu_backward
AMSA/mmsr/models/archs/DCNv2/src/cpu/dcn_v2_cpu.cpp:26
Functiondcn_v2_cpu_forward
AMSA/mmsr/models/archs/DCNv2/src/cpu/dcn_v2_cpu.cpp:7
Functiondcn_v2_forward
AMSA/mmsr/models/archs/DCNv2/src/dcn_v2.h:9
Functiondcn_v2_psroi_pooling_backward
AMSA/mmsr/models/archs/DCNv2/src/dcn_v2.h:109
Functiondcn_v2_psroi_pooling_cpu_backward
AMSA/mmsr/models/archs/DCNv2/src/cpu/dcn_v2_cpu.cpp:58
Functiondcn_v2_psroi_pooling_cpu_forward
AMSA/mmsr/models/archs/DCNv2/src/cpu/dcn_v2_cpu.cpp:42
Functiondcn_v2_psroi_pooling_forward
AMSA/mmsr/models/archs/DCNv2/src/dcn_v2.h:75
Functiondefine_net_d
(opt)
AMSA/mmsr/models/networks.py:36
Functiondefine_net_extractor
(opt)
AMSA/mmsr/models/networks.py:52
Functiondefine_net_g
(opt)
AMSA/mmsr/models/networks.py:27
Functiondefine_net_map
(opt)
AMSA/mmsr/models/networks.py:44
Functiondefine_net_student
(opt)
AMSA/mmsr/models/networks.py:61
Functiondefine_net_teacher
(opt)
AMSA/mmsr/models/networks.py:70
Functiondeform_conv_backward_input_cuda
AMSA/mmsr/models/archs/dcn/src/deform_conv_cuda.cpp:260
Functiondeform_conv_backward_parameters_cuda
AMSA/mmsr/models/archs/dcn/src/deform_conv_cuda.cpp:373
Functiondeform_conv_forward_cuda
AMSA/mmsr/models/archs/dcn/src/deform_conv_cuda.cpp:151
Functiondict_constructor
(loader, node)
AMSA/mmsr/utils/options.py:24
Functiondict_representer
(dumper, data)
AMSA/mmsr/utils/options.py:21
Methoddist_validation
(self, dataloader, current_iter, tb_logger, save_img)
AMSA/mmsr/models/teacher_contras_model.py:231
Methoddist_validation
(self, dataloader, current_iter, tb_logger, save_img)
AMSA/mmsr/models/student_contras_distillation_model.py:282
Methodfeed_data
(self, data)
AMSA/mmsr/models/base_model.py:31
Functionflipx4_forward
Flip 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
Functionflow_warp
Warp an image or feature map with optical flow. Args: x (Tensor): Tensor with size (N, C, H, W). flow (Tensor): Tensor with size
AMSA/mmsr/models/archs/arch_util.py:190
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, pred, mask=None)
AMSA/mmsr/models/losses.py:130
Methodforward
(self, x, gt)
AMSA/mmsr/models/losses.py:197
Methodforward
(self, x, gt)
AMSA/mmsr/models/losses.py:253
Methodforward
Args: input (Tensor): The input for the loss module, i.e., the network prediction. target_is_real (bo
AMSA/mmsr/models/losses.py:340
Methodforward
Args: discriminator (nn.Module): Network for the discriminator. real_data (Tensor): Real input data. fake
AMSA/mmsr/models/losses.py:414
Methodforward
Args: x (Tensor): The input for the loss module. maps (Tensor): The maps generated by swap module. weight
AMSA/mmsr/models/losses.py:475
Methodforward
(self, swapped_features, gt)
AMSA/mmsr/models/losses.py:581
Methodforward
(self, x)
AMSA/mmsr/models/archs/vgg_arch.py:135
Methodforward
(self, x)
AMSA/mmsr/models/archs/arch_util.py:132
Methodforward
(self, x)
AMSA/mmsr/models/archs/arch_util.py:184
Methodforward
(self, x)
AMSA/mmsr/models/archs/ref_restoration_arch.py:24
Methodforward
Args: x (Tensor): the input image of SRNTT. maps (dict[Tensor]): the swapped feature maps on relu3_1, relu2_1
AMSA/mmsr/models/archs/ref_restoration_arch.py:59
Methodforward
(self, x, pre_offset_list, img_ref_list, max_val_list)
AMSA/mmsr/models/archs/ref_restoration_arch.py:184
Methodforward
(self, batch)
AMSA/mmsr/models/archs/contras_extractor_arch.py:38
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, x)
AMSA/mmsr/models/archs/discriminator_arch.py:31
Methodforward
(self, dense_features, img_ref_hr)
AMSA/mmsr/models/archs/corres_generation_arch.py:49
Methodforward
(self, x, inref_y, inref_x)
AMSA/mmsr/models/archs/patch_match_arch.py:24
Methodforward
(self, input_features, ref_features, aggregated_x, aggregated_y, is_test=False)
AMSA/mmsr/models/archs/patch_match_arch.py:41
Methodforward
(self, inref_x, inref_y, dilation)
AMSA/mmsr/models/archs/patch_match_arch.py:74
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, weight, stride=1,
AMSA/mmsr/models/archs/dcn/deform_conv.py:18
Methodforward
(ctx, input, offset, mask, weight,
AMSA/mmsr/models/archs/dcn/deform_conv.py:119
Methodforward
(self, x, offset)
AMSA/mmsr/models/archs/dcn/deform_conv.py:237
Methodforward
(self, x)
AMSA/mmsr/models/archs/dcn/deform_conv.py:261
Methodforward
(self, x, offset, mask)
AMSA/mmsr/models/archs/dcn/deform_conv.py:308
Methodforward
(self, x)
AMSA/mmsr/models/archs/dcn/deform_conv.py:334
Methodforward
(ctx, input, offset, mask, weight, bias, stride, padding, dilation, deformable_groups)
AMSA/mmsr/models/archs/DCNv2/dcn_v2.py:19
Methodforward
(self, input, offset, mask)
AMSA/mmsr/models/archs/DCNv2/dcn_v2.py:88
Methodforward
(self, input)
AMSA/mmsr/models/archs/DCNv2/dcn_v2.py:126
Methodforward
(self, x)
AMSA/mmsr/models/archs/DCNv2/dcn_v2.py:167
Methodforward
Args: pre_offset: precomputed_offset. Size: [b, 9, h, w, 2]
AMSA/mmsr/models/archs/DCNv2/dcn_v2.py:222
Methodforward
Args: pre_offset: precomputed_offset. Size: [b, 9, h, w, 2]
AMSA/mmsr/models/archs/DCNv2/dcn_v2.py:291
Methodforward
(ctx, input, rois, offset, spatial_scale,
AMSA/mmsr/models/archs/DCNv2/dcn_v2.py:328
Methodforward
(self, input, rois, offset)
AMSA/mmsr/models/archs/DCNv2/dcn_v2.py:403
Methodforward
(self, input, rois)
AMSA/mmsr/models/archs/DCNv2/dcn_v2.py:443
Methodget
(self, filepath)
AMSA/mmsr/utils/file_client.py:16
Methodget
(self, filepath)
AMSA/mmsr/utils/file_client.py:35
Methodget
(self, filepath)
AMSA/mmsr/utils/file_client.py:72
Methodget
Get values according to the filepath from one lmdb named client_key. Args: filepath (str | obj:`Path`): Here, filepath is the lmd
AMSA/mmsr/utils/file_client.py:135
Methodget
(self, filepath)
AMSA/mmsr/utils/file_client.py:157
Methodget_current_visuals
(self)
AMSA/mmsr/models/base_model.py:37
Methodget_lr
(self)
AMSA/mmsr/models/lr_scheduler.py:35
Methodget_lr
(self)
AMSA/mmsr/models/lr_scheduler.py:94
Methodget_text
(self, filepath)
AMSA/mmsr/utils/file_client.py:20
Methodget_text
(self, filepath)
AMSA/mmsr/utils/file_client.py:41
Methodget_text
(self, filepath)
AMSA/mmsr/utils/file_client.py:79
Methodget_text
(self, filepath)
AMSA/mmsr/utils/file_client.py:150
Methodget_text
(self, filepath)
AMSA/mmsr/utils/file_client.py:163
Methodget_text
(self, filepath)
AMSA/mmsr/utils/file_client.py:217
Functionimresize
(img, scale, antialiasing=True)
AMSA/mmsr/data/util.py:400
Functionimresize_np
(img, scale, antialiasing=True)
AMSA/mmsr/data/util.py:475
Functionindex_generation
Generate an index list for reading num_frames frames from a sequence of images. Args: crt_idx (int): Current center index. ma
AMSA/mmsr/data/util.py:31
Functioninit_func
(m)
AMSA/mmsr/models/archs/arch_util.py:13
Methodinit_training_settings
(self)
AMSA/mmsr/models/ref_restoration_model.py:90
Functionmake_layer
Make layers by stacking the same blocks. Args: basic_block (nn.module): nn.module class for basic block. n_basic_blocks (int): nu
AMSA/mmsr/models/archs/arch_util.py:63
Functionmasked_loss
Create a masked version of a given loss function. To use this decorator, the loss function must have the signature like `loss_func(pred, targ
AMSA/mmsr/models/loss_utils.py:58
Functionmodulated_deform_conv_cuda_backward
AMSA/mmsr/models/archs/dcn/src/deform_conv_cuda.cpp:566
Functionmodulated_deform_conv_cuda_forward
AMSA/mmsr/models/archs/dcn/src/deform_conv_cuda.cpp:486
Methodnondist_validation
(self, dataloader, current_iter, tb_logger, save_img)
AMSA/mmsr/models/ref_restoration_model.py:361
Methodnondist_validation
(self, dataloader, current_iter, tb_logger, save_img)
AMSA/mmsr/models/teacher_contras_model.py:235
Methodnondist_validation
(self, dataloader, current_iter, tb_logger, save_img)
AMSA/mmsr/models/student_contras_distillation_model.py:286
Methodoptimize_parameters
(self)
AMSA/mmsr/models/base_model.py:34
Methodoptimize_parameters
(self, step)
AMSA/mmsr/models/ref_restoration_model.py:197
Methodoptimize_parameters
(self, step)
AMSA/mmsr/models/teacher_contras_model.py:212
Methodoptimize_parameters
(self, step)
AMSA/mmsr/models/student_contras_distillation_model.py:259
Functionpaired_random_crop
Paired random crop. It crops lists of lq and gt images with corresponding locations. Args: img_gts (list[ndarray] | ndarray): GT ima
AMSA/mmsr/data/transforms.py:27
Functionpixel_unshuffle
Pixel unshuffle. Args: x (Tensor): the input feature. The shape is [b, c, hh, hw]. s (int): downsample ratio. Returns:
AMSA/mmsr/models/archs/arch_util.py:270
Functionpsnr
Calculate PSNR (Peak Signal-to-Noise Ratio). Ref: https://en.wikipedia.org/wiki/Peak_signal-to-noise_ratio Args: img1, img2 (ndarray
AMSA/mmsr/utils/metrics.py:34
← previousnext →201–300 of 312, ranked by callers