Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Zj-BinXia/AMSA
/ functions
Functions
312 in github.com/Zj-BinXia/AMSA
⨍
Functions
312
◇
Types & classes
61
↓ 1 callers
Method
init_training_settings
(self)
AMSA/mmsr/models/teacher_contras_model.py:85
↓ 1 callers
Method
init_training_settings
(self)
AMSA/mmsr/models/student_contras_distillation_model.py:96
↓ 1 callers
Function
insert_bn
Inserts bn layer after each conv. Args: names (list): The list of layer names. Returns: list: The list of layer names with b
AMSA/mmsr/models/archs/vgg_arch.py:40
↓ 1 callers
Function
l1_loss
(pred, target)
AMSA/mmsr/models/losses.py:17
↓ 1 callers
Function
main
()
AMSA/mmsr/train.py:20
↓ 1 callers
Function
main
()
AMSA/mmsr/test.py:13
↓ 1 callers
Function
make_cuda_ext
(name, module, sources)
AMSA/setup.py:89
↓ 1 callers
Function
make_cuda_ext
(name, sources)
AMSA/mmsr/models/archs/dcn/setup.py:5
↓ 1 callers
Function
mask_reduce_loss
Apply element-wise weight and reduce loss. Args: loss (Tensor): Element-wise loss. weight (Tensor): Element-wise weights. Default
AMSA/mmsr/models/loss_utils.py:26
↓ 1 callers
Function
mse_loss
(pred, target)
AMSA/mmsr/models/losses.py:22
↓ 1 callers
Method
nondist_validation
(self, dataloader, current_iter, tb_logger, save_img)
AMSA/mmsr/models/sr_model.py:164
↓ 1 callers
Method
optimize_parameters
(self, step)
AMSA/mmsr/models/sr_model.py:94
↓ 1 callers
Function
ordered_yaml
Support OrderedDict for yaml. Returns: yaml Loader and Dumper.
AMSA/mmsr/utils/options.py:8
↓ 1 callers
Method
re_init_dcn_offset
(self)
AMSA/mmsr/models/archs/ref_restoration_arch.py:43
↓ 1 callers
Function
read_cufed_img_worker
(path, key, compress_level, is_ref)
AMSA/mmsr/scripts/create_cufed_lmdb.py:228
↓ 1 callers
Function
readme
()
AMSA/setup.py:11
↓ 1 callers
Function
reduce_loss
Reduce loss as specified. Args: loss (Tensor): Elementwise loss tensor. reduction (str): Options are "none", "mean" and "sum".
AMSA/mmsr/models/loss_utils.py:6
↓ 1 callers
Method
reset_parameters
(self)
AMSA/mmsr/models/archs/dcn/deform_conv.py:230
↓ 1 callers
Method
reset_parameters
(self)
AMSA/mmsr/models/archs/dcn/deform_conv.py:299
↓ 1 callers
Method
reset_parameters
(self)
AMSA/mmsr/models/archs/DCNv2/dcn_v2.py:80
↓ 1 callers
Method
resume_training
Reload the optimizers and schedulers for resumed training. Args: resume_state (dict): Resume state.
AMSA/mmsr/models/base_model.py:292
↓ 1 callers
Method
set_epoch
(self, epoch)
AMSA/mmsr/data/data_sampler.py:68
↓ 1 callers
Function
set_random_seed
Set random seeds.
AMSA/mmsr/utils/util.py:75
↓ 1 callers
Method
setup_optimizers
(self)
AMSA/mmsr/models/sr_model.py:73
↓ 1 callers
Method
setup_optimizers
(self)
AMSA/mmsr/models/teacher_contras_model.py:93
↓ 1 callers
Method
setup_optimizers
(self)
AMSA/mmsr/models/student_contras_distillation_model.py:104
↓ 1 callers
Function
srntt_init_weights
(net, init_type='normal', init_gain=0.02)
AMSA/mmsr/models/archs/arch_util.py:12
↓ 1 callers
Method
start
(self)
AMSA/mmsr/utils/util.py:276
↓ 1 callers
Method
test
(self)
AMSA/mmsr/models/sr_model.py:117
↓ 1 callers
Method
test
(self)
AMSA/mmsr/models/ref_restoration_model.py:309
↓ 1 callers
Method
test
(self)
AMSA/mmsr/models/teacher_contras_model.py:225
↓ 1 callers
Method
test
(self)
AMSA/mmsr/models/student_contras_distillation_model.py:274
↓ 1 callers
Method
update_learning_rate
Update learning rate. Args: current_iter (int): Current iteration. warmup_iter (int)： Warmup iter numbers. -1 for no
AMSA/mmsr/models/base_model.py:155
↓ 1 callers
Function
upscale_positions
(pos, scaling_steps=0)
AMSA/mmsr/models/teacher_contras_model.py:30
↓ 1 callers
Function
upscale_positions
(pos, scaling_steps=0)
AMSA/mmsr/models/student_contras_distillation_model.py:30
↓ 1 callers
Function
warp
(pos1, max_h, max_w, transformed_coordinates)
AMSA/mmsr/models/teacher_contras_model.py:42
↓ 1 callers
Function
warp
(pos1, max_h, max_w, transformed_coordinates)
AMSA/mmsr/models/student_contras_distillation_model.py:42
↓ 1 callers
Function
write_version_py
()
AMSA/setup.py:69
Function
DUF_downsample
Downsamping with Gaussian kernel used in the DUF official code Args: x (Tensor, [B, T, C, H, W]): frames to be downsampled. scale
AMSA/mmsr/utils/util.py:166
Function
PYBIND11_MODULE
AMSA/mmsr/models/archs/dcn/src/deform_conv_cuda.cpp:681
Function
PYBIND11_MODULE
AMSA/mmsr/models/archs/DCNv2/src/vision.cpp:4
Method
__call__
Format logging message. Args: log_vars (dict): It contains the following keys: epoch (int): Epoch number.
AMSA/mmsr/utils/logger.py:32
Method
__getitem__
(self, index)
AMSA/mmsr/data/contras_dataset.py:149
Method
__getitem__
(self, index)
AMSA/mmsr/data/contras_dataset.py:297
Method
__getitem__
(self, index)
AMSA/mmsr/data/ref_cufed_dataset.py:68
Method
__init__
(self, opt, start_iter=1, tb_logger=None)
AMSA/mmsr/utils/logger.py:21
Method
__init__
(self)
AMSA/mmsr/utils/file_client.py:27
Method
__init__
(self, server_list_cfg, client_cfg, sys_path=None)
AMSA/mmsr/utils/file_client.py:55
Method
__init__
(self, db_paths, client_keys='default', readonly=True,
AMSA/mmsr/utils/file_client.py:103
Method
__init__
(self, backend='disk', **kwargs)
AMSA/mmsr/utils/file_client.py:190
Method
__init__
(self, task_num=0, bar_width=50, start=True)
AMSA/mmsr/utils/util.py:257
Method
__init__
(self, opt)
AMSA/mmsr/data/contras_dataset.py:261
Method
__init__
(self, dataset, num_replicas=None, rank=None, ratio=100)
AMSA/mmsr/data/data_sampler.py:31
Method
__init__
(self, opt)
AMSA/mmsr/data/ref_cufed_dataset.py:40
Method
__init__
(self, opt)
AMSA/mmsr/models/sr_model.py:24
Method
__init__
(self, opt)
AMSA/mmsr/models/base_model.py:18
Method
__init__
(self, opt)
AMSA/mmsr/models/ref_restoration_model.py:22
Method
__init__
(self, loss_weight=1.0, reduction='mean')
AMSA/mmsr/models/losses.py:69
Method
__init__
(self, loss_weight=1.0, reduction='mean', eps=1e-12)
AMSA/mmsr/models/losses.py:104
Method
__init__
(self, loss_weight=1.0)
AMSA/mmsr/models/losses.py:127
Method
__init__
(self, layer_weights, vgg_type='vgg19', use_input_norm=True
AMSA/mmsr/models/losses.py:167
Method
__init__
(self, gan_type, real_label_val=1.0, fake_label_val=0.0,
AMSA/mmsr/models/losses.py:287
Method
__init__
(self, loss_weight=1.)
AMSA/mmsr/models/losses.py:410
Method
__init__
(self, use_weights=False, loss_weight=1.0, vgg_type='vgg19'
AMSA/mmsr/models/losses.py:447
Method
__init__
(self, vgg_type='vgg19', map_weight=1.0, criterion='l1')
AMSA/mmsr/models/losses.py:562
Method
__init__
(self, optimizer, period, restarts=[0], re
AMSA/mmsr/models/lr_scheduler.py:72
Method
__init__
(self, opt)
AMSA/mmsr/models/teacher_contras_model.py:69
Method
__init__
(self, opt)
AMSA/mmsr/models/student_contras_distillation_model.py:69
Method
__init__
(self, layer_name_list, vgg_type='vgg19', use_input_norm=Tr
AMSA/mmsr/models/archs/vgg_arch.py:80
Method
__init__
(self, nf=64, res_scale=1, pytorch_init=False,
AMSA/mmsr/models/archs/arch_util.py:98
Method
__init__
(self, nf=64, bn_affine=True)
AMSA/mmsr/models/archs/arch_util.py:174
Method
__init__
(self, in_nc=3, out_nc=3, nf=64, n_blocks=16)
AMSA/mmsr/models/archs/ref_restoration_arch.py:11
Method
__init__
(self, ngf=64, n_blocks=16, groups=8)
AMSA/mmsr/models/archs/ref_restoration_arch.py:77
Method
__init__
(self)
AMSA/mmsr/models/archs/contras_extractor_arch.py:10
Method
__init__
(self, in_nc=3, ndf=32)
AMSA/mmsr/models/archs/discriminator_arch.py:8
Method
__init__
(self, patch_size=3, stride=1, vgg_layer_list=['relu3_1', '
AMSA/mmsr/models/archs/corres_generation_arch.py:16
Method
__init__
Args: modulation (bool, optional): If True, Modulated Defomable Convolution (Deformable ConvNets v2).
AMSA/mmsr/models/archs/patch_match_arch.py:9
Method
__init__
(self, filter_size)
AMSA/mmsr/models/archs/patch_match_arch.py:66
Method
__init__
(self)
AMSA/mmsr/models/archs/patch_match_arch.py:106
Method
__init__
(self)
AMSA/mmsr/models/archs/patch_match_arch.py:163
Method
__init__
(self, *args, **kwargs)
AMSA/mmsr/models/archs/dcn/deform_conv.py:244
Method
__init__
(self, in_channels, out_channels, kernel_size,
AMSA/mmsr/models/archs/dcn/deform_conv.py:269
Method
__init__
(self, *args, extra_offset_mask=False, **kwargs)
AMSA/mmsr/models/archs/dcn/deform_conv.py:316
Method
__init__
(self, in_channels, out_channels, kernel_size,
AMSA/mmsr/models/archs/DCNv2/dcn_v2.py:58
Method
__init__
(self, in_channels, out_channels, kernel_size,
AMSA/mmsr/models/archs/DCNv2/dcn_v2.py:139
Method
__init__
(self, in_channels, out_channels, kernel_size,
AMSA/mmsr/models/archs/DCNv2/dcn_v2.py:194
Method
__init__
(self, in_channels, out_channels, kernel_size,
AMSA/mmsr/models/archs/DCNv2/dcn_v2.py:263
Method
__init__
(self, spatial_scale, pooled_size, output_dim,
AMSA/mmsr/models/archs/DCNv2/dcn_v2.py:384
Method
__init__
(self, spatial_scale, pooled_size, output_dim,
AMSA/mmsr/models/archs/DCNv2/dcn_v2.py:415
Method
__iter__
(self)
AMSA/mmsr/data/data_sampler.py:50
Method
__len__
(self)
AMSA/mmsr/data/contras_dataset.py:230
Method
__len__
(self)
AMSA/mmsr/data/contras_dataset.py:374
Method
__len__
(self)
AMSA/mmsr/data/data_sampler.py:65
Method
__len__
(self)
AMSA/mmsr/data/ref_cufed_dataset.py:208
Method
__missing__
(self, key)
AMSA/mmsr/utils/options.py:125
Method
_transform
(v, op)
AMSA/mmsr/models/sr_model.py:128
Method
_wgan_loss
wgan loss. Args: input (Tensor): Input tensor. target (bool): Target label. Returns: Tensor: wga
AMSA/mmsr/models/losses.py:310
Method
backward
(ctx, grad_output)
AMSA/mmsr/models/archs/dcn/deform_conv.py:63
Method
backward
(ctx, grad_output)
AMSA/mmsr/models/archs/dcn/deform_conv.py:155
Method
backward
(ctx, grad_output)
AMSA/mmsr/models/archs/DCNv2/dcn_v2.py:358
← previous
next →
101–200 of 312, ranked by callers