MCPcopy Create free account

hub / github.com/ZengxiZhang/HUPE / functions

Functions239 in github.com/ZengxiZhang/HUPE

Method__init__
(self, opt)
code/models/base_model.py:14
Method__init__
(self, opt, step)
code/models/LLFlow_model.py:18
Method__init__
(self, optimizer, T_period, restarts=None, weights=None, eta_min=0, last_epoch=-1)
code/models/lr_scheduler.py:38
Method__init__
(self, _pretrained_=True)
code/models/modules/loss.py:12
Method__init__
norm (bool): normalize/denormalize the stats
code/models/modules/loss.py:35
Method__init__
(self, device='0', vgg=None, weights=None, indices=None, normalize=True)
code/models/modules/loss.py:51
Method__init__
(self, device='0', vgg=None, weights=None, indices=None, normalize=True)
code/models/modules/loss.py:78
Method__init__
(self, eps=1e-6)
code/models/modules/loss.py:111
Method__init__
(self, gan_type, real_label_val=1.0, fake_label_val=0.0)
code/models/modules/loss.py:122
Method__init__
(self, device=torch.device('cpu'))
code/models/modules/loss.py:156
Method__init__
(self, channels, activation='lrelu', kernel=3)
code/models/modules/base_layers.py:42
Method__init__
(self, in_channels, out_channels, activation='lrelu')
code/models/modules/base_layers.py:58
Method__init__
(self, in_channels, out_channels, activation='lrelu')
code/models/modules/base_layers.py:69
Method__init__
(self, in_channels, out_channels, activation='lrelu', stride=1)
code/models/modules/base_layers.py:92
Method__init__
(self, in_channels, out_channels, activation='lrelu')
code/models/modules/base_layers.py:107
Method__init__
(self, kernel_size=2, stride=2)
code/models/modules/base_layers.py:122
Method__init__
(self, kernel_size=2, stride=2)
code/models/modules/base_layers.py:131
Method__init__
(self, in_planes, ratio=16)
code/models/modules/base_layers.py:140
Method__init__
(self, kernel_size=3)
code/models/modules/base_layers.py:157
Method__init__
(self, planes)
code/models/modules/base_layers.py:171
Method__init__
(self, in_channels, out_channels, kernel_size=[3, 3], stride=[1, 1], padding
code/models/modules/flow.py:56
Method__init__
(self, factor)
code/models/modules/flow.py:153
Method__init__
(self, in_planes, out_planes, kernel_size, stride=1, padding=0, dilation=1, groups=1, relu=True, bn=True, bias
code/models/modules/cbam_fromdehazeflow.py:7
Method__init__
(self, gate_channels, reduction_ratio=16, pool_types=('avg', 'max'))
code/models/modules/cbam_fromdehazeflow.py:27
Method__init__
(self)
code/models/modules/cbam_fromdehazeflow.py:78
Method__init__
(self, nf=64, gc=32, bias=True)
code/models/modules/color_encoder.py:14
Method__init__
(self, nf, opt=None)
code/models/modules/color_encoder.py:53
Method__init__
(self, nf=64, gc=32, bias=True)
code/models/modules/RRDBNet_arch.py:13
Method__init__
(self, in_nc, out_nc, nf, nb, gc=32, scale=4, opt=None)
code/models/modules/RRDBNet_arch.py:52
Method__init__
(self, nf=64, gc=32, bias=True)
code/models/modules/ConditionEncoder.py:14
Method__init__
(self, in_nc, out_nc, nf, nb, gc=32, scale=4, opt=None)
code/models/modules/ConditionEncoder.py:63
Method__init__
(self, in_nc, out_nc, nf, nb, gc=32, scale=4, opt=None)
code/models/modules/ConditionEncoder.py:154
Method__init__
(self, nf=64)
code/models/modules/module_util.py:43
Method__init__
(self, in_channels, hidden_channels, actnorm_scale=1.0, flow_permutation="invconv", flow_coup
code/models/modules/FlowStep.py:33
Method__init__
(self, num_channels, LU_decomposed=False)
code/models/modules/Permutations.py:13
Method__init__
(self, in_channels, opt)
code/models/modules/FlowAffineCouplingsAblation.py:12
Method__init__
(self, in_channels=24, kernel_size=3)
code/models/modules/FlowAffineCouplingsAblation.py:195
Method__init__
(self, in_size, out_size, downsample, relu_slope, use_csff=False)
code/models/modules/FlowAffineCouplingsAblation.py:236
Method__init__
(self, image_shape, hidden_channels, K, L=None, actnorm_scale=1.0, flow_perm
code/models/modules/FlowUpsamplerNet.py:17
Method__init__
(self, num_channels, logs_eps=0, cond_channels=0, position=None, consume_ratio=0.5, opt=None)
code/models/modules/Split.py:14
Method__init__
(self, num_features, scale=1.)
code/models/modules/FlowActNorms.py:114
Method__init__
(self, num_features, scale=1.)
code/models/modules/FlowActNorms.py:126
Method__init__
(self, requires_grad=False)
code/models/modules/vgg.py:42
Method__init__
(self, in_nc, out_nc, nf, nb, gc=32, scale=4, K=None, opt=None, step=None)
code/models/modules/LLFlow_arch.py:22
Method__missing__
(self, key)
code/options/options.py:30
Method_check_input_dim
(self, input)
code/models/modules/FlowActNorms.py:117
Functioncalc_style_loss
(input, target)
code/models/modules/loss.py:190
Functioncalculate_psnr
(img1, img2)
code/utils/util.py:217
Functioncat_feature
(tensor_a, tensor_b)
code/models/modules/thops.py:52
Functioncheck_resume
Check resume states and pretrain_model paths
code/options/options.py:47
Functioncompute_psnr_ssim
(recoverd, clean)
code/utils/util.py:269
Functiondefine_Flow
(opt, step)
code/models/networks.py:28
Functiondict2str
dict to string for logger
code/options/options.py:16
Functiondict_constructor
(loader, node)
code/utils/util.py:112
Functiondict_representer
(dumper, data)
code/utils/util.py:109
Functiondict_to_nonedict
(opt)
code/options/options.py:35
Methodencode
(self, gt, rrdbResults, logdet=0.0, epses=None, y_onehot=None)
code/models/modules/FlowUpsamplerNet.py:207
Functionflow_warp
Warp an image or feature map with optical flow Args: x (Tensor): size (N, C, H, W) flow (Tensor): size (N, H, W, 2), normal value
code/models/modules/module_util.py:58
Methodforward
(self, img1, img2)
code/utils/util.py:68
Methodforward
(self, pred, true, layer='conv5_2')
code/models/modules/loss.py:29
Methodforward
(self, x, y)
code/models/modules/loss.py:115
Methodforward
(self, input, target_is_real)
code/models/modules/loss.py:150
Methodforward
(self, interp, interp_crit)
code/models/modules/loss.py:166
Methodforward
(self, img1, img2)
code/models/modules/loss.py:235
Methodforward
(self, R, I_att)
code/models/modules/base_layers.py:22
Methodforward
(self, x)
code/models/modules/base_layers.py:53
Methodforward
(self, x)
code/models/modules/base_layers.py:65
Methodforward
(self, x)
code/models/modules/base_layers.py:81
Methodforward
(self, x)
code/models/modules/base_layers.py:102
Methodforward
(self, x)
code/models/modules/base_layers.py:117
Methodforward
(self, x)
code/models/modules/base_layers.py:126
Methodforward
(self, x)
code/models/modules/base_layers.py:135
Methodforward
(self, x)
code/models/modules/base_layers.py:150
Methodforward
(self, x)
code/models/modules/base_layers.py:162
Methodforward
(self, x)
code/models/modules/base_layers.py:175
Methodforward
(self, x, y)
code/models/modules/base_layers.py:182
Methodforward
(self, input)
code/models/modules/flow.py:68
Methodforward
(self, input, logdet=None, reverse=False)
code/models/modules/flow.py:157
Methodforward
(self, x)
code/models/modules/cbam_fromdehazeflow.py:14
Methodforward
(self, x)
code/models/modules/cbam_fromdehazeflow.py:23
Methodforward
(self, x)
code/models/modules/cbam_fromdehazeflow.py:37
Methodforward
(self, x)
code/models/modules/cbam_fromdehazeflow.py:74
Methodforward
(self, x)
code/models/modules/cbam_fromdehazeflow.py:83
Methodforward
(self, x)
code/models/modules/cbam_fromdehazeflow.py:96
Methodforward
(self, x)
code/models/modules/color_encoder.py:27
Methodforward
(self, x)
code/models/modules/color_encoder.py:45
Methodforward
(self, x, get_steps=False)
code/models/modules/color_encoder.py:77
Methodforward
(self, x)
code/models/modules/RRDBNet_arch.py:26
Methodforward
(self, x)
code/models/modules/RRDBNet_arch.py:44
Methodforward
(self, x, get_steps=False)
code/models/modules/RRDBNet_arch.py:76
Methodforward
(self, x)
code/models/modules/ConditionEncoder.py:27
Methodforward
(self, x)
code/models/modules/ConditionEncoder.py:48
Methodforward
(self, x, get_steps=False)
code/models/modules/ConditionEncoder.py:89
Methodforward
(self, x, get_steps=False)
code/models/modules/ConditionEncoder.py:188
Methodforward
(self, x)
code/models/modules/module_util.py:51
Methodforward
(self, input, logdet=None, reverse=False, rrdbResults=None)
code/models/modules/FlowStep.py:70
Methodforward
log-det = log|abs(|W|)| * pixels
code/models/modules/Permutations.py:45
Methodforward
(self, input: torch.Tensor, logdet=None, reverse=False, ft=None)
code/models/modules/FlowAffineCouplingsAblation.py:73
Methodforward
(self, input_x)
code/models/modules/FlowAffineCouplingsAblation.py:201
Methodforward
(self, x, pha, amp)
code/models/modules/FlowAffineCouplingsAblation.py:218
← previousnext →101–200 of 239, ranked by callers