MCPcopy Create free account

hub / github.com/HeZongyao/LMF / functions

Functions212 in github.com/HeZongyao/LMF

Method__init__
(self, encoder_spec, imnet_spec=None, hidden_dim=256)
models/lte.py:20
Method__init__
(self, encoder, prenet_q, hypernet_q, imnet_q, imnet_k, imnet_v, local_size=
models/lmciaosr.py:16
Method__init__
(self, dim, window_size, num_heads, qkv_bias=True, qk_scale=None, attn_drop=0., proj_drop=0.)
models/swinir.py:83
Method__init__
(self, dim, input_resolution, num_heads, window_size=7, shift_size=0, mlp_ratio=4., qkv_bias=
models/swinir.py:187
Method__init__
(self, input_resolution, dim, norm_layer=nn.LayerNorm)
models/swinir.py:313
Method__init__
(self, dim, input_resolution, depth, num_heads, window_size, mlp_ratio=4., qkv_bias=True, qk_
models/swinir.py:373
Method__init__
(self, dim, input_resolution, depth, num_heads, window_size, mlp_ratio=4., qkv_bias=True, qk_
models/swinir.py:446
Method__init__
(self, img_size=224, patch_size=4, in_chans=3, embed_dim=96, norm_layer=None)
models/swinir.py:510
Method__init__
(self, img_size=224, patch_size=4, in_chans=3, embed_dim=96, norm_layer=None)
models/swinir.py:553
Method__init__
(self, scale, num_feat)
models/swinir.py:584
Method__init__
(self, scale, num_feat, num_out_ch, input_resolution=None)
models/swinir.py:608
Method__init__
(self, img_size=48, patch_size=1, in_chans=3, embed_dim=180, depths=[6, 6, 6, 6, 6, 6], num_h
models/swinir.py:650
Method__init__
(self, rgb_range, rgb_mean=(0.4488, 0.4371, 0.4040), rgb_std=(1.0, 1.0, 1.0), sign=-1)
models/edsr.py:22
Method__init__
(self, conv, n_feats, kernel_size, bias=True, bn=False, act=nn.ReLU(True), res_scale=1)
models/edsr.py:33
Method__init__
(self, conv, scale, n_feats, bn=False, act=False, bias=True)
models/edsr.py:55
Method__init__
(self, in_dim, out_dim, hidden_list, use_conv=False)
models/mlp.py:9
Method__init__
(self, encoder, imnet_q, imnet_k, imnet_v,
models/ciaosr.py:30
Method__init__
(self, inChannels, growRate, kSize=3)
models/rdn.py:14
Method__init__
(self, args)
models/rdn.py:46
Method__init__
(self, in_dim, out_dim, hidden_dim, hidden_depth, mod_scale=True, mod_shift=True, mod_up_merg
models/lmmlp.py:10
Method__init__
(self, encoder_spec, num_layer=3, hidden_dim=256, out_dim=3)
models/ltep.py:18
Method__init__
(self, layer_name_list, vgg_type='vgg19', use_input_norm=Tr
models/arch_ciaosr/vgg_arch.py:80
Method__init__
(self)
models/arch_ciaosr/arch_csnln.py:92
Method__init__
(self)
models/arch_ciaosr/arch_csnln.py:127
Method__init__
(self, patch_size=3, stride=1, vgg_layer_list=['relu3_1'],
models/arch_ciaosr/arch_csnln.py:250
Method__init__
(self, patch_size=3, stride=1)
models/arch_ciaosr/arch_csnln.py:320
Method__init__
(self, in_nc=3, out_nc=3, nf=64, n_blocks=16)
models/arch_ciaosr/arch_csnln.py:385
Method__init__
(self, channel=64, reduction=2, ksize=3, scale=2, stride=1, softmax_scale=10, average=True, conv=default_conv)
models/arch_ciaosr/arch_csnln.py:407
Method__len__
(self)
datasets/image_folder.py:56
Method__len__
(self)
datasets/image_folder.py:83
Method__len__
(self)
datasets/wrappers.py:24
Method__len__
(self)
datasets/wrappers.py:88
Method__len__
(self)
datasets/wrappers.py:168
Method__len__
(self)
datasets/wrappers.py:239
Method__len__
(self)
datasets/wrappers.py:318
Method_init_weights
(self, m)
models/swinir.py:773
Methodaugment
(x)
datasets/wrappers.py:50
Methodaugment
(x)
datasets/wrappers.py:116
Methodaugment
(x)
datasets/wrappers.py:194
Methodaugment
(x)
datasets/wrappers.py:269
Functioncalc_psnr
(sr, hr, dataset=None, scale=1, rgb_range=1)
utils.py:136
Functioncenter_crop
Crop all images in a folder to a specified size, centering the crop area. :param folder: Directory containing images to be cropped. :par
scripts/resize.py:25
Functioncompute_num_params
(model, text=False)
utils.py:83
Functioncontinuous_resize
Continuously resize images in a folder across multiple scale steps and crop to maintain a specific dimension. :param folder: Directory conta
scripts/resize.py:72
Functiondecorator
(cls)
datasets/datasets.py:8
Functiondecorator
(cls)
models/models.py:8
Functiondefault_conv
(in_channels, out_channels, kernel_size, bias=True)
models/rcan.py:10
Functiondefault_conv
(in_channels, out_channels, kernel_size, bias=True)
models/edsr.py:15
Functiondefault_conv
(in_channels, out_channels, kernel_size,stride=1, bias=True)
models/arch_ciaosr/arch_csnln.py:11
Methodextra_repr
(self)
models/swinir.py:151
Methodextra_repr
(self)
models/swinir.py:285
Methodextra_repr
(self)
models/swinir.py:343
Methodextra_repr
(self)
models/swinir.py:411
Methodflops
(self, N)
models/swinir.py:154
Methodflops
(self)
models/swinir.py:289
Methodflops
(self)
models/swinir.py:346
Methodflops
(self)
models/swinir.py:414
Methodflops
(self)
models/swinir.py:534
Methodflops
(self)
models/swinir.py:571
Methodflops
(self)
models/swinir.py:616
Methodflops
(self)
models/swinir.py:851
Methodforward
(self, x)
models/rcan.py:62
Methodforward
(self, x)
models/rcan.py:84
Methodforward
(self, x)
models/rcan.py:103
Methodforward
(self, x)
models/rcan.py:154
Methodforward
Forward function. :param inp: Input image (B, h * w, 3) :param coord: Coordinates (B, H * W, 2) :param cell: Cell ar
models/lmlte.py:624
Methodforward
(self, inp, coord, cell)
models/misc.py:73
Methodforward
(self, inp, coord, cell, bsize=None)
models/liif.py:146
Methodforward
Forward function. :param inp: Input image (B, h * w, 3) :param coord: Coordinates (B, H * W, 2) :param cell: Cell ar
models/lmliif.py:567
Methodforward
(self, inp, coord, cell, bsize=None)
models/lte.py:145
Methodforward
Forward function. :param inp: Input image (B, h * w, 3) :param coord: Coordinates (B, H * W, 2) :param cell: Cell ar
models/lmciaosr.py:618
Methodforward
(self, x)
models/swinir.py:28
Methodforward
Args: x: input features with shape of (num_windows*B, N, C) mask: (0/-inf) mask with shape of (num_windows, Wh*Ww, Wh
models/swinir.py:118
Methodforward
(self, x, x_size)
models/swinir.py:243
Methodforward
x: B, H*W, C
models/swinir.py:320
Methodforward
(self, x, x_size)
models/swinir.py:401
Methodforward
(self, x, x_size)
models/swinir.py:485
Methodforward
(self, x)
models/swinir.py:528
Methodforward
(self, x, x_size)
models/swinir.py:566
Methodforward
(self, x)
models/swinir.py:812
Methodforward
(self, x)
models/edsr.py:47
Methodforward
(self, x)
models/edsr.py:136
Methodforward
(self, x)
models/mlp.py:26
Methodforward
Forward function. Args: x: input tensor. coord (Tensor): coordinates tensor. cell (Tensor): cell tensor.
models/ciaosr.py:269
Methodforward
(self, x)
models/rdn.py:23
Methodforward
(self, x)
models/rdn.py:42
Methodforward
(self, x)
models/rdn.py:98
Methodforward
(self, x, mod=None, coord=None, only_layer0=False, skip_layer0=False)
models/lmmlp.py:42
Methodforward
(self, inp, coord, cell)
models/ltep.py:110
Methodforward
(self, x)
models/arch_ciaosr/vgg_arch.py:135
Methodforward
(self, batch)
models/arch_ciaosr/arch_csnln.py:119
Methodforward
(self, image1, image2)
models/arch_ciaosr/arch_csnln.py:133
Methodforward
(self, dense_features, img_ref_hr)
models/arch_ciaosr/arch_csnln.py:282
Methodforward
(self, feats_in, feats_ref)
models/arch_ciaosr/arch_csnln.py:346
Methodforward
(self, x)
models/arch_ciaosr/arch_csnln.py:398
Methodforward
(self, input)
models/arch_ciaosr/arch_csnln.py:429
Methodload_state_dict
(self, state_dict, strict=True)
models/edsr.py:150
Functionmake
(dataset_spec, args=None)
datasets/datasets.py:14
Functionmake
(model_spec, args=None, load_sd=False, strict=True)
models/models.py:14
Functionmake_edsr
(n_resblocks=32, n_feats=256, res_scale=0.1, scale=2, no_upsampling=False, rgb_range=1)
models/edsr.py:187
← previousnext →101–200 of 212, ranked by callers