SYMBOL INDEX (1327 symbols across 168 files) FILE: Deraining/Deraining/Metric/PSNR.py function main (line 12) | def main(): FILE: Deraining/Deraining/utils.py function calculate_psnr (line 10) | def calculate_psnr(img1, img2, border=0): function calculate_ssim (line 31) | def calculate_ssim(img1, img2, border=0): function ssim (line 58) | def ssim(img1, img2): function load_img (line 80) | def load_img(filepath): function save_img (line 83) | def save_img(filepath, img): function load_gray_img (line 86) | def load_gray_img(filepath): function save_gray_img (line 89) | def save_gray_img(filepath, img): FILE: Deraining/basicsr/data/__init__.py function create_dataset (line 29) | def create_dataset(dataset_opt): function create_dataloader (line 56) | def create_dataloader(dataset, function worker_init_fn (line 122) | def worker_init_fn(worker_id, num_workers, rank, seed): FILE: Deraining/basicsr/data/data_sampler.py class EnlargedSampler (line 6) | class EnlargedSampler(Sampler): method __init__ (line 21) | def __init__(self, dataset, num_replicas, rank, ratio=1): method __iter__ (line 30) | def __iter__(self): method __len__ (line 45) | def __len__(self): method set_epoch (line 48) | def set_epoch(self, epoch): FILE: Deraining/basicsr/data/data_util.py function read_img_seq (line 12) | def read_img_seq(path, require_mod_crop=False, scale=1): function generate_frame_indices (line 36) | def generate_frame_indices(crt_idx, function paired_paths_from_lmdb (line 92) | def paired_paths_from_lmdb(folders, keys): function paired_paths_from_meta_info_file (line 158) | def paired_paths_from_meta_info_file(folders, keys, meta_info_file, function paired_paths_from_folder (line 208) | def paired_paths_from_folder(folders, keys, filename_tmpl): function paired_DP_paths_from_folder (line 253) | def paired_DP_paths_from_folder(folders, keys, filename_tmpl): function paths_from_folder (line 307) | def paths_from_folder(folder): function paths_from_lmdb (line 322) | def paths_from_lmdb(folder): function generate_gaussian_kernel (line 338) | def generate_gaussian_kernel(kernel_size=13, sigma=1.6): function duf_downsample (line 356) | def duf_downsample(x, kernel_size=13, scale=4): FILE: Deraining/basicsr/data/ffhq_dataset.py class FFHQDataset (line 9) | class FFHQDataset(data.Dataset): method __init__ (line 22) | def __init__(self, opt): method __getitem__ (line 46) | def __getitem__(self, index): method __len__ (line 64) | def __len__(self): FILE: Deraining/basicsr/data/paired_image_dataset.py class Dataset_PairedImage (line 16) | class Dataset_PairedImage(data.Dataset): method __init__ (line 45) | def __init__(self, opt): method __getitem__ (line 78) | def __getitem__(self, index): method __len__ (line 131) | def __len__(self): class Dataset_GaussianDenoising (line 134) | class Dataset_GaussianDenoising(data.Dataset): method __init__ (line 162) | def __init__(self, opt): method __getitem__ (line 198) | def __getitem__(self, index): method __len__ (line 273) | def __len__(self): class Dataset_DefocusDeblur_DualPixel_16bit (line 276) | class Dataset_DefocusDeblur_DualPixel_16bit(data.Dataset): method __init__ (line 277) | def __init__(self, opt): method __getitem__ (line 299) | def __getitem__(self, index): method __len__ (line 362) | def __len__(self): FILE: Deraining/basicsr/data/prefetch_dataloader.py class PrefetchGenerator (line 7) | class PrefetchGenerator(threading.Thread): method __init__ (line 18) | def __init__(self, generator, num_prefetch_queue): method run (line 25) | def run(self): method __next__ (line 30) | def __next__(self): method __iter__ (line 36) | def __iter__(self): class PrefetchDataLoader (line 40) | class PrefetchDataLoader(DataLoader): method __init__ (line 55) | def __init__(self, num_prefetch_queue, **kwargs): method __iter__ (line 59) | def __iter__(self): class CPUPrefetcher (line 63) | class CPUPrefetcher(): method __init__ (line 70) | def __init__(self, loader): method next (line 74) | def next(self): method reset (line 80) | def reset(self): class CUDAPrefetcher (line 84) | class CUDAPrefetcher(): method __init__ (line 97) | def __init__(self, loader, opt): method preload (line 105) | def preload(self): method next (line 118) | def next(self): method reset (line 124) | def reset(self): FILE: Deraining/basicsr/data/reds_dataset.py class REDSDataset (line 12) | class REDSDataset(data.Dataset): method __init__ (line 51) | def __init__(self, opt): method __getitem__ (line 106) | def __getitem__(self, index): method __len__ (line 236) | def __len__(self): FILE: Deraining/basicsr/data/single_image_dataset.py class SingleImageDataset (line 9) | class SingleImageDataset(data.Dataset): method __init__ (line 25) | def __init__(self, opt): method __getitem__ (line 48) | def __getitem__(self, index): method __len__ (line 66) | def __len__(self): FILE: Deraining/basicsr/data/transforms.py function mod_crop (line 5) | def mod_crop(img, scale): function paired_random_crop (line 24) | def paired_random_crop(img_gts, img_lqs, lq_patch_size, scale, gt_path): function paired_random_crop_DP (line 85) | def paired_random_crop_DP(img_lqLs, img_lqRs, img_gts, gt_patch_size, sc... function augment (line 136) | def augment(imgs, hflip=True, rotation=True, flows=None, return_status=F... function img_rotate (line 203) | def img_rotate(img, angle, center=None, scale=1.0): function data_augmentation (line 223) | def data_augmentation(image, mode): function random_augmentation (line 270) | def random_augmentation(*args): FILE: Deraining/basicsr/data/video_test_dataset.py class VideoTestDataset (line 11) | class VideoTestDataset(data.Dataset): method __init__ (line 44) | def __init__(self, opt): method __getitem__ (line 117) | def __getitem__(self, index): method __len__ (line 146) | def __len__(self): class VideoTestVimeo90KDataset (line 150) | class VideoTestVimeo90KDataset(data.Dataset): method __init__ (line 170) | def __init__(self, opt): method __getitem__ (line 211) | def __getitem__(self, index): method __len__ (line 227) | def __len__(self): class VideoTestDUFDataset (line 231) | class VideoTestDUFDataset(VideoTestDataset): method __getitem__ (line 244) | def __getitem__(self, index): class VideoRecurrentTestDataset (line 293) | class VideoRecurrentTestDataset(VideoTestDataset): method __init__ (line 304) | def __init__(self, opt): method __getitem__ (line 309) | def __getitem__(self, index): method __len__ (line 324) | def __len__(self): FILE: Deraining/basicsr/data/vimeo90k_dataset.py class Vimeo90KDataset (line 10) | class Vimeo90KDataset(data.Dataset): method __init__ (line 50) | def __init__(self, opt): method __getitem__ (line 78) | def __getitem__(self, index): method __len__ (line 129) | def __len__(self): FILE: Deraining/basicsr/metrics/fid.py function load_patched_inception_v3 (line 10) | def load_patched_inception_v3(device='cuda', function extract_inception_features (line 23) | def extract_inception_features(data_generator, function calculate_fid (line 57) | def calculate_fid(mu1, sigma1, mu2, sigma2, eps=1e-6): FILE: Deraining/basicsr/metrics/metric_util.py function reorder_image (line 6) | def reorder_image(img, input_order='HWC'): function to_y_channel (line 34) | def to_y_channel(img): FILE: Deraining/basicsr/metrics/niqe.py function estimate_aggd_param (line 10) | def estimate_aggd_param(block): function compute_feature (line 40) | def compute_feature(block): function niqe (line 67) | def niqe(img, function calculate_niqe (line 158) | def calculate_niqe(img, crop_border, input_order='HWC', convert_to='y'): FILE: Deraining/basicsr/metrics/psnr_ssim.py function calculate_psnr (line 9) | def calculate_psnr(img1, function _ssim (line 66) | def _ssim(img1, img2): function prepare_for_ssim (line 101) | def prepare_for_ssim(img, k): function prepare_for_ssim_rgb (line 115) | def prepare_for_ssim_rgb(img, k): function _3d_gaussian_calculator (line 131) | def _3d_gaussian_calculator(img, conv3d): function _generate_3d_gaussian_kernel (line 135) | def _generate_3d_gaussian_kernel(): function _ssim_3d (line 145) | def _ssim_3d(img1, img2, max_value): function _ssim_cly (line 184) | def _ssim_cly(img1, img2): function calculate_ssim (line 225) | def calculate_ssim(img1, FILE: Deraining/basicsr/models/__init__.py function create_model (line 21) | def create_model(opt): FILE: Deraining/basicsr/models/archs/__init__.py function dynamic_instantiation (line 21) | def dynamic_instantiation(modules, cls_type, opt): function define_network (line 43) | def define_network(opt): FILE: Deraining/basicsr/models/archs/arch_util.py function default_init_weights (line 19) | def default_init_weights(module_list, scale=1, bias_fill=0, **kwargs): function make_layer (line 49) | def make_layer(basic_block, num_basic_block, **kwarg): class ResidualBlockNoBN (line 65) | class ResidualBlockNoBN(nn.Module): method __init__ (line 80) | def __init__(self, num_feat=64, res_scale=1, pytorch_init=False): method forward (line 90) | def forward(self, x): class Upsample (line 96) | class Upsample(nn.Sequential): method __init__ (line 104) | def __init__(self, scale, num_feat): function flow_warp (line 119) | def flow_warp(x, function resize_flow (line 164) | def resize_flow(flow, function pixel_unshuffle (line 211) | def pixel_unshuffle(x, scale): FILE: Deraining/basicsr/models/archs/common.py function default_conv (line 7) | def default_conv(in_channels, out_channels, kernel_size, bias=True): class ResBlock (line 10) | class ResBlock(nn.Module): method __init__ (line 11) | def __init__( method forward (line 27) | def forward(self, x): class MeanShift (line 33) | class MeanShift(nn.Conv2d): method __init__ (line 34) | def __init__(self, rgb_range, rgb_mean, rgb_std, sign=-1): class Upsampler (line 45) | class Upsampler(nn.Sequential): method __init__ (line 46) | def __init__(self, conv, scale, n_feat, act=False, bias=True): FILE: Deraining/basicsr/models/archs/mamber32_arch.py class SelectiveScanFn (line 20) | class SelectiveScanFn(torch.autograd.Function): method forward (line 23) | def forward(ctx, u, delta, A, B, C, D=None, delta_bias=None, delta_sof... method backward (line 60) | def backward(ctx, dout, *args): function selective_scan_fn_v1 (line 88) | def selective_scan_fn_v1(u, delta, A, B, C, D=None, delta_bias=None, del... function flops_selective_scan_fn (line 98) | def flops_selective_scan_fn(B=1, L=256, D=768, N=16, with_D=True, with_Z... function print_jit_input_names (line 121) | def print_jit_input_names(inputs): function selective_scan_flop_jit (line 130) | def selective_scan_flop_jit(inputs, outputs): function to_3d (line 141) | def to_3d(x): function to_4d (line 144) | def to_4d(x,h,w): class BiasFree_LayerNorm (line 147) | class BiasFree_LayerNorm(nn.Module): method __init__ (line 148) | def __init__(self, normalized_shape): method forward (line 159) | def forward(self, x): class WithBias_LayerNorm (line 163) | class WithBias_LayerNorm(nn.Module): method __init__ (line 164) | def __init__(self, normalized_shape): method forward (line 176) | def forward(self, x): class LayerNorm (line 182) | class LayerNorm(nn.Module): method __init__ (line 183) | def __init__(self, dim, LayerNorm_type): method forward (line 190) | def forward(self, x): class FeedForward (line 198) | class FeedForward(nn.Module): method __init__ (line 199) | def __init__(self, dim, ffn_expansion_factor, bias): method forward (line 210) | def forward(self, x): class SS2D_1 (line 219) | class SS2D_1(nn.Module): method __init__ (line 220) | def __init__( method dt_init (line 335) | def dt_init(dt_rank, d_inner, dt_scale=1.0, dt_init="random", dt_min=0... method A_log_init (line 361) | def A_log_init(d_state, d_inner, copies=-1, device=None, merge=True): method D_init (line 379) | def D_init(d_inner, copies=-1, device=None, merge=True): method forward_corev1 (line 392) | def forward_corev1(self, x: torch.Tensor): method cforward_corev1 (line 436) | def cforward_corev1(self, xc: torch.Tensor): method forward (line 483) | def forward(self, x: torch.Tensor, **kwargs): class MamberBlock (line 498) | class MamberBlock(nn.Module): method __init__ (line 499) | def __init__(self, dim, num_heads, ffn_expansion_factor, bias, LayerNo... method forward (line 507) | def forward(self, x): class OverlapPatchEmbed (line 516) | class OverlapPatchEmbed(nn.Module): method __init__ (line 517) | def __init__(self, in_c=3, embed_dim=48, bias=False): method forward (line 522) | def forward(self, x): class Downsample (line 531) | class Downsample(nn.Module): method __init__ (line 532) | def __init__(self, n_feat): method forward (line 538) | def forward(self, x): class Upsample (line 541) | class Upsample(nn.Module): method __init__ (line 542) | def __init__(self, n_feat): method forward (line 548) | def forward(self, x): class Mamber32 (line 554) | class Mamber32(nn.Module): method __init__ (line 555) | def __init__(self, method forward (line 610) | def forward(self, inp_img): method flops (line 653) | def flops(self, shape=(3, 64, 64)): FILE: Deraining/basicsr/models/archs/mamber33_arch.py class SelectiveScanFn (line 20) | class SelectiveScanFn(torch.autograd.Function): method forward (line 23) | def forward(ctx, u, delta, A, B, C, D=None, delta_bias=None, delta_sof... method backward (line 60) | def backward(ctx, dout, *args): function selective_scan_fn_v1 (line 87) | def selective_scan_fn_v1(u, delta, A, B, C, D=None, delta_bias=None, del... function flops_selective_scan_fn (line 95) | def flops_selective_scan_fn(B=1, L=256, D=768, N=16, with_D=True, with_Z... function print_jit_input_names (line 118) | def print_jit_input_names(inputs): function selective_scan_flop_jit (line 127) | def selective_scan_flop_jit(inputs, outputs): function to_3d (line 138) | def to_3d(x): function to_4d (line 141) | def to_4d(x,h,w): class BiasFree_LayerNorm (line 144) | class BiasFree_LayerNorm(nn.Module): method __init__ (line 145) | def __init__(self, normalized_shape): method forward (line 156) | def forward(self, x): class WithBias_LayerNorm (line 160) | class WithBias_LayerNorm(nn.Module): method __init__ (line 161) | def __init__(self, normalized_shape): method forward (line 173) | def forward(self, x): class LayerNorm (line 179) | class LayerNorm(nn.Module): method __init__ (line 180) | def __init__(self, dim, LayerNorm_type): method forward (line 187) | def forward(self, x): class FeedForward (line 195) | class FeedForward(nn.Module): method __init__ (line 196) | def __init__(self, dim, ffn_expansion_factor, bias): method forward (line 207) | def forward(self, x): class SS2D_1 (line 216) | class SS2D_1(nn.Module): method __init__ (line 217) | def __init__( method dt_init (line 332) | def dt_init(dt_rank, d_inner, dt_scale=1.0, dt_init="random", dt_min=0... method A_log_init (line 357) | def A_log_init(d_state, d_inner, copies=-1, device=None, merge=True): method D_init (line 374) | def D_init(d_inner, copies=-1, device=None, merge=True): method forward_corev1 (line 386) | def forward_corev1(self, x: torch.Tensor): method cforward_corev1 (line 430) | def cforward_corev1(self, xc: torch.Tensor): method forward (line 480) | def forward(self, x: torch.Tensor, **kwargs): class MamberBlock (line 496) | class MamberBlock(nn.Module): method __init__ (line 497) | def __init__(self, dim, num_heads, ffn_expansion_factor, bias, LayerNo... method forward (line 505) | def forward(self, x): class OverlapPatchEmbed (line 514) | class OverlapPatchEmbed(nn.Module): method __init__ (line 515) | def __init__(self, in_c=3, embed_dim=48, bias=False): method forward (line 520) | def forward(self, x): class Downsample (line 529) | class Downsample(nn.Module): method __init__ (line 530) | def __init__(self, n_feat): method forward (line 536) | def forward(self, x): class Upsample (line 539) | class Upsample(nn.Module): method __init__ (line 540) | def __init__(self, n_feat): method forward (line 546) | def forward(self, x): class Mamber33 (line 552) | class Mamber33(nn.Module): method __init__ (line 553) | def __init__(self, method forward (line 608) | def forward(self, inp_img): method flops (line 651) | def flops(self, shape=(3, 64, 64)): FILE: Deraining/basicsr/models/archs/restormer_arch.py function to_3d (line 21) | def to_3d(x): function to_4d (line 24) | def to_4d(x,h,w): class BiasFree_LayerNorm (line 27) | class BiasFree_LayerNorm(nn.Module): method __init__ (line 28) | def __init__(self, normalized_shape): method forward (line 39) | def forward(self, x): class WithBias_LayerNorm (line 43) | class WithBias_LayerNorm(nn.Module): method __init__ (line 44) | def __init__(self, normalized_shape): method forward (line 56) | def forward(self, x): class LayerNorm (line 62) | class LayerNorm(nn.Module): method __init__ (line 63) | def __init__(self, dim, LayerNorm_type): method forward (line 70) | def forward(self, x): class FeedForward (line 78) | class FeedForward(nn.Module): method __init__ (line 79) | def __init__(self, dim, ffn_expansion_factor, bias): method forward (line 90) | def forward(self, x): class Attention (line 101) | class Attention(nn.Module): method __init__ (line 102) | def __init__(self, dim, num_heads, bias): method forward (line 113) | def forward(self, x): class TransformerBlock (line 139) | class TransformerBlock(nn.Module): method __init__ (line 140) | def __init__(self, dim, num_heads, ffn_expansion_factor, bias, LayerNo... method forward (line 148) | def forward(self, x): class OverlapPatchEmbed (line 158) | class OverlapPatchEmbed(nn.Module): method __init__ (line 159) | def __init__(self, in_c=3, embed_dim=48, bias=False): method forward (line 164) | def forward(self, x): class Downsample (line 173) | class Downsample(nn.Module): method __init__ (line 174) | def __init__(self, n_feat): method forward (line 180) | def forward(self, x): class Upsample (line 183) | class Upsample(nn.Module): method __init__ (line 184) | def __init__(self, n_feat): method forward (line 190) | def forward(self, x): class Restormer (line 195) | class Restormer(nn.Module): method __init__ (line 196) | def __init__(self, method forward (line 247) | def forward(self, inp_img): method flops (line 288) | def flops(self, shape=(3, 256, 256)): FILE: Deraining/basicsr/models/base_model.py class BaseModel (line 14) | class BaseModel(): method __init__ (line 17) | def __init__(self, opt): method feed_data (line 24) | def feed_data(self, data): method optimize_parameters (line 27) | def optimize_parameters(self): method get_current_visuals (line 30) | def get_current_visuals(self): method save (line 33) | def save(self, epoch, current_iter): method validation (line 37) | def validation(self, dataloader, current_iter, tb_logger, save_img=Fal... method model_ema (line 54) | def model_ema(self, decay=0.999): method get_current_log (line 64) | def get_current_log(self): method model_to_device (line 67) | def model_to_device(self, net): method setup_schedulers (line 87) | def setup_schedulers(self): method get_bare_model (line 135) | def get_bare_model(self, net): method print_network (line 144) | def print_network(self, net): method _set_lr (line 164) | def _set_lr(self, lr_groups_l): method _get_init_lr (line 174) | def _get_init_lr(self): method update_learning_rate (line 183) | def update_learning_rate(self, current_iter, warmup_iter=-1): method get_current_learning_rate (line 207) | def get_current_learning_rate(self): method save_network (line 214) | def save_network(self, net, net_label, current_iter, param_key='params'): method _print_different_keys_loading (line 246) | def _print_different_keys_loading(self, crt_net, load_net, strict=True): method load_network (line 281) | def load_network(self, net, load_path, strict=True, param_key='params'): method save_training_state (line 312) | def save_training_state(self, epoch, current_iter): method resume_training (line 336) | def resume_training(self, resume_state): method reduce_loss_dict (line 353) | def reduce_loss_dict(self, loss_dict): FILE: Deraining/basicsr/models/image_restoration_model.py class Mixing_Augment (line 22) | class Mixing_Augment: method __init__ (line 23) | def __init__(self, mixup_beta, use_identity, device): method mixup (line 31) | def mixup(self, target, input_): method __call__ (line 41) | def __call__(self, target, input_): class ImageCleanModel (line 51) | class ImageCleanModel(BaseModel): method __init__ (line 54) | def __init__(self, opt): method init_training_settings (line 78) | def init_training_settings(self): method setup_optimizers (line 115) | def setup_optimizers(self): method feed_train_data (line 136) | def feed_train_data(self, data): method feed_data (line 144) | def feed_data(self, data): method optimize_parameters (line 149) | def optimize_parameters(self, current_iter): method pad_test (line 175) | def pad_test(self, window_size): method nonpad_test (line 188) | def nonpad_test(self, img=None): method dist_validation (line 207) | def dist_validation(self, dataloader, current_iter, tb_logger, save_im... method nondist_validation (line 213) | def nondist_validation(self, dataloader, current_iter, tb_logger, method _log_validation_metric_values (line 300) | def _log_validation_metric_values(self, current_iter, dataset_name, method get_current_visuals (line 311) | def get_current_visuals(self): method save (line 319) | def save(self, epoch, current_iter): FILE: Deraining/basicsr/models/losses/loss_util.py function reduce_loss (line 5) | def reduce_loss(loss, reduction): function weight_reduce_loss (line 25) | def weight_reduce_loss(loss, weight=None, reduction='mean'): function weighted_loss (line 57) | def weighted_loss(loss_func): FILE: Deraining/basicsr/models/losses/losses.py function l1_loss (line 12) | def l1_loss(pred, target): function mse_loss (line 17) | def mse_loss(pred, target): class L1Loss (line 26) | class L1Loss(nn.Module): method __init__ (line 35) | def __init__(self, loss_weight=1.0, reduction='mean'): method forward (line 44) | def forward(self, pred, target, weight=None, **kwargs): class MSELoss (line 55) | class MSELoss(nn.Module): method __init__ (line 64) | def __init__(self, loss_weight=1.0, reduction='mean'): method forward (line 73) | def forward(self, pred, target, weight=None, **kwargs): class PSNRLoss (line 84) | class PSNRLoss(nn.Module): method __init__ (line 86) | def __init__(self, loss_weight=1.0, reduction='mean', toY=False): method forward (line 95) | def forward(self, pred, target): class CharbonnierLoss (line 111) | class CharbonnierLoss(nn.Module): method __init__ (line 114) | def __init__(self, loss_weight=1.0, reduction='mean', eps=1e-3): method forward (line 118) | def forward(self, x, y): FILE: Deraining/basicsr/models/lr_scheduler.py class MultiStepRestartLR (line 7) | class MultiStepRestartLR(_LRScheduler): method __init__ (line 20) | def __init__(self, method get_lr (line 35) | def get_lr(self): class LinearLR (line 49) | class LinearLR(_LRScheduler): method __init__ (line 59) | def __init__(self, method get_lr (line 66) | def get_lr(self): class VibrateLR (line 72) | class VibrateLR(_LRScheduler): method __init__ (line 82) | def __init__(self, method get_lr (line 89) | def get_lr(self): function get_position_from_periods (line 115) | def get_position_from_periods(iteration, cumulative_period): class CosineAnnealingRestartLR (line 136) | class CosineAnnealingRestartLR(_LRScheduler): method __init__ (line 156) | def __init__(self, method get_lr (line 172) | def get_lr(self): class CosineAnnealingRestartCyclicLR (line 186) | class CosineAnnealingRestartCyclicLR(_LRScheduler): method __init__ (line 203) | def __init__(self, method get_lr (line 219) | def get_lr(self): FILE: Deraining/basicsr/test.py function main (line 13) | def main(): FILE: Deraining/basicsr/train.py function parse_options (line 24) | def parse_options(is_train=True): function init_loggers (line 61) | def init_loggers(opt): function create_train_val_dataloader (line 82) | def create_train_val_dataloader(opt, logger): function main (line 131) | def main(): FILE: Deraining/basicsr/utils/bundle_submissions.py function bundle_submissions_raw (line 13) | def bundle_submissions_raw(submission_folder,session): function bundle_submissions_srgb (line 46) | def bundle_submissions_srgb(submission_folder,session): function bundle_submissions_srgb_v1 (line 79) | def bundle_submissions_srgb_v1(submission_folder,session): FILE: Deraining/basicsr/utils/create_lmdb.py function prepare_keys (line 7) | def prepare_keys(folder_path, suffix='png'): function create_lmdb_for_reds (line 24) | def create_lmdb_for_reds(): function create_lmdb_for_gopro (line 46) | def create_lmdb_for_gopro(): function create_lmdb_for_rain13k (line 71) | def create_lmdb_for_rain13k(): function create_lmdb_for_SIDD (line 84) | def create_lmdb_for_SIDD(): FILE: Deraining/basicsr/utils/dist_util.py function init_dist (line 10) | def init_dist(launcher, backend='nccl', **kwargs): function _init_dist_pytorch (line 21) | def _init_dist_pytorch(backend, **kwargs): function _init_dist_slurm (line 28) | def _init_dist_slurm(backend, port=None): function get_dist_info (line 61) | def get_dist_info(): function master_only (line 75) | def master_only(func): FILE: Deraining/basicsr/utils/download_util.py function download_file_from_google_drive (line 8) | def download_file_from_google_drive(file_id, save_path): function get_confirm_token (line 41) | def get_confirm_token(response): function save_response_content (line 48) | def save_response_content(response, FILE: Deraining/basicsr/utils/face_util.py class FaceRestorationHelper (line 16) | class FaceRestorationHelper(object): method __init__ (line 19) | def __init__(self, upscale_factor, face_size=512): method init_dlib (line 41) | def init_dlib(self, detection_path, landmark5_path, landmark68_path): method free_dlib_gpu_memory (line 47) | def free_dlib_gpu_memory(self): method read_input_image (line 52) | def read_input_image(self, img_path): method detect_faces (line 56) | def detect_faces(self, method get_face_landmarks_5 (line 89) | def get_face_landmarks_5(self): method get_face_landmarks_68 (line 96) | def get_face_landmarks_68(self): method warp_crop_faces (line 132) | def warp_crop_faces(self, method add_restored_face (line 170) | def add_restored_face(self, face): method paste_faces_to_input_image (line 173) | def paste_faces_to_input_image(self, save_path): method clean_all (line 211) | def clean_all(self): FILE: Deraining/basicsr/utils/file_client.py class BaseStorageBackend (line 5) | class BaseStorageBackend(metaclass=ABCMeta): method get (line 14) | def get(self, filepath): method get_text (line 18) | def get_text(self, filepath): class MemcachedBackend (line 22) | class MemcachedBackend(BaseStorageBackend): method __init__ (line 32) | def __init__(self, server_list_cfg, client_cfg, sys_path=None): method get (line 49) | def get(self, filepath): method get_text (line 56) | def get_text(self, filepath): class HardDiskBackend (line 60) | class HardDiskBackend(BaseStorageBackend): method get (line 63) | def get(self, filepath): method get_text (line 69) | def get_text(self, filepath): class LmdbBackend (line 76) | class LmdbBackend(BaseStorageBackend): method __init__ (line 96) | def __init__(self, method get (line 131) | def get(self, filepath, client_key): method get_text (line 146) | def get_text(self, filepath): class FileClient (line 150) | class FileClient(object): method __init__ (line 169) | def __init__(self, backend='disk', **kwargs): method get (line 177) | def get(self, filepath, client_key='default'): method get_text (line 185) | def get_text(self, filepath): FILE: Deraining/basicsr/utils/flow_util.py function flowread (line 7) | def flowread(flow_path, quantize=False, concat_axis=0, *args, **kwargs): function flowwrite (line 47) | def flowwrite(flow, filename, quantize=False, concat_axis=0, *args, **kw... function quantize_flow (line 78) | def quantize_flow(flow, max_val=0.02, norm=True): function dequantize_flow (line 106) | def dequantize_flow(dx, dy, max_val=0.02, denorm=True): function quantize (line 130) | def quantize(arr, min_val, max_val, levels, dtype=np.int64): function dequantize (line 157) | def dequantize(arr, min_val, max_val, levels, dtype=np.float64): FILE: Deraining/basicsr/utils/img_util.py function img2tensor (line 9) | def img2tensor(imgs, bgr2rgb=True, float32=True): function tensor2img (line 36) | def tensor2img(tensor, rgb2bgr=True, out_type=np.uint8, min_max=(0, 1)): function imfrombytes (line 101) | def imfrombytes(content, flag='color', float32=False): function imfrombytesDP (line 127) | def imfrombytesDP(content, flag='color', float32=False): function padding (line 148) | def padding(img_lq, img_gt, gt_size): function padding_DP (line 166) | def padding_DP(img_lqL, img_lqR, img_gt, gt_size): function imwrite (line 181) | def imwrite(img, file_path, params=None, auto_mkdir=True): function crop_border (line 200) | def crop_border(imgs, crop_border): FILE: Deraining/basicsr/utils/lmdb_util.py function make_lmdb_from_imgs (line 9) | def make_lmdb_from_imgs(data_path, function read_img_worker (line 139) | def read_img_worker(path, key, compress_level): class LmdbMaker (line 164) | class LmdbMaker(): method __init__ (line 175) | def __init__(self, method put (line 194) | def put(self, img_byte, key, img_shape): method close (line 205) | def close(self): FILE: Deraining/basicsr/utils/logger.py class MessageLogger (line 10) | class MessageLogger(): method __init__ (line 23) | def __init__(self, opt, start_iter=1, tb_logger=None): method __call__ (line 34) | def __call__(self, log_vars): function init_tb_logger (line 81) | def init_tb_logger(log_dir): function init_wandb_logger (line 88) | def init_wandb_logger(opt): function get_root_logger (line 108) | def get_root_logger(logger_name='basicsr', log_level=logging.INFO, log_f... function get_env_info (line 150) | def get_env_info(): FILE: Deraining/basicsr/utils/matlab_functions.py function cubic (line 6) | def cubic(x): function calculate_weights_indices (line 17) | def calculate_weights_indices(in_length, out_length, scale, kernel, function imresize (line 88) | def imresize(img, scale, antialiasing=True): function rgb2ycbcr (line 173) | def rgb2ycbcr(img, y_only=False): function bgr2ycbcr (line 207) | def bgr2ycbcr(img, y_only=False): function ycbcr2rgb (line 241) | def ycbcr2rgb(img): function ycbcr2bgr (line 273) | def ycbcr2bgr(img): function _convert_input_type_range (line 305) | def _convert_input_type_range(img): function _convert_output_type_range (line 333) | def _convert_output_type_range(img, dst_type): FILE: Deraining/basicsr/utils/misc.py function set_random_seed (line 12) | def set_random_seed(seed): function get_time_str (line 21) | def get_time_str(): function mkdir_and_rename (line 25) | def mkdir_and_rename(path): function make_exp_dirs (line 39) | def make_exp_dirs(opt): function scandir (line 53) | def scandir(dir_path, suffix=None, recursive=False, full_path=False): function scandir_SIDD (line 95) | def scandir_SIDD(dir_path, keywords=None, recursive=False, full_path=Fal... function check_resume (line 137) | def check_resume(opt, resume_iter): function sizeof_fmt (line 166) | def sizeof_fmt(size, suffix='B'): FILE: Deraining/basicsr/utils/options.py function ordered_yaml (line 6) | def ordered_yaml(): function parse (line 31) | def parse(opt_path, is_train=True): function dict2str (line 92) | def dict2str(opt, indent_level=1): FILE: Deraining/basicsr/utils2.py function calculate_psnr (line 10) | def calculate_psnr(img1, img2, border=0): function calculate_ssim (line 31) | def calculate_ssim(img1, img2, border=0): function ssim (line 58) | def ssim(img1, img2): function load_img (line 80) | def load_img(filepath): function save_img (line 83) | def save_img(filepath, img): function load_gray_img (line 86) | def load_gray_img(filepath): function save_gray_img (line 89) | def save_gray_img(filepath, img): FILE: Deraining/setup.py function readme (line 16) | def readme(): function get_git_hash (line 23) | def get_git_hash(): function get_hash (line 49) | def get_hash(): function write_version_py (line 64) | def write_version_py(): function get_version (line 84) | def get_version(): function make_cuda_ext (line 90) | def make_cuda_ext(name, module, sources, sources_cuda=None): function get_requirements (line 116) | def get_requirements(filename='requirements.txt'): FILE: Mamba/kernels/selective_scan/csrc/selective_scan/cus/selective_scan.cpp function set_ssm_params_fwd (line 36) | void set_ssm_params_fwd(SSMParamsBase ¶ms, function set_ssm_params_bwd (line 98) | void set_ssm_params_bwd(SSMParamsBwd ¶ms, function selective_scan_fwd (line 157) | std::vector function selective_scan_bwd (line 241) | std::vector function PYBIND11_MODULE (line 351) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) { FILE: Mamba/kernels/selective_scan/csrc/selective_scan/cusndstate/selective_scan_ndstate.cpp function set_ssm_params_fwd (line 36) | void set_ssm_params_fwd(SSMParamsBase ¶ms, function set_ssm_params_bwd (line 93) | void set_ssm_params_bwd(SSMParamsBwd ¶ms, function selective_scan_fwd (line 148) | std::vector function selective_scan_bwd (line 230) | std::vector function PYBIND11_MODULE (line 338) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) { FILE: Mamba/kernels/selective_scan/csrc/selective_scan/cusndstate/selective_scan_ndstate.h type SSMScanParamsBase (line 9) | struct SSMScanParamsBase { type SSMParamsBase (line 26) | struct SSMParamsBase { function SSMParamsBase (line 60) | struct SSMParamsBwd: public SSMParamsBase { FILE: Mamba/kernels/selective_scan/csrc/selective_scan/cusnrow/selective_scan_nrow.cpp function set_ssm_params_fwd (line 44) | void set_ssm_params_fwd(SSMParamsBase ¶ms, function set_ssm_params_bwd (line 106) | void set_ssm_params_bwd(SSMParamsBwd ¶ms, function selective_scan_fwd (line 165) | std::vector function selective_scan_bwd (line 251) | std::vector function PYBIND11_MODULE (line 364) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) { FILE: Mamba/kernels/selective_scan/csrc/selective_scan/cusoflex/selective_scan_oflex.cpp function set_ssm_params_fwd (line 36) | void set_ssm_params_fwd(SSMParamsBase ¶ms, function set_ssm_params_bwd (line 98) | void set_ssm_params_bwd(SSMParamsBwd ¶ms, function selective_scan_fwd (line 157) | std::vector function selective_scan_bwd (line 245) | std::vector function PYBIND11_MODULE (line 360) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) { FILE: Mamba/kernels/selective_scan/csrc/selective_scan/selective_scan.h type SSMScanParamsBase (line 9) | struct SSMScanParamsBase { type SSMParamsBase (line 26) | struct SSMParamsBase { function SSMParamsBase (line 63) | struct SSMParamsBwd: public SSMParamsBase { FILE: Mamba/kernels/selective_scan/csrc/selective_scan/selective_scan_common.h type BytesToType (line 29) | struct BytesToType type BytesToType (line 34) | struct BytesToType type BytesToType (line 39) | struct BytesToType type BytesToType (line 44) | struct BytesToType type BytesToType (line 49) | struct BytesToType function __device__ (line 58) | static inline __device__ void to_float(const scalar_t (&src)[N], float (... function __device__ (line 66) | static inline __device__ void to_float(const at::Half (&src)[N], float (... function __device__ (line 78) | static inline __device__ void to_float(const at::BFloat16 (&src)[N], flo... function float (line 92) | struct SSMScanOp { function __device__ (line 104) | __device__ SSMScanPrefixCallbackOp(scan_t running_prefix_) : running_pre... function __device__ (line 107) | __device__ scan_t operator()(scan_t block_aggregate) { function load_input (line 117) | void load_input(typename Ktraits::input_t *u, function load_weight (line 134) | void load_weight(typename Ktraits::input_t *Bvar, function store_output (line 156) | void store_output(typename Ktraits::input_t *out, function store_output1 (line 176) | void store_output1(typename Ktraits::output_t *out, function load_output (line 196) | void load_output(typename Ktraits::output_t *u, FILE: Mamba/kernels/selective_scan/setup.py function get_cuda_bare_metal_version (line 30) | def get_cuda_bare_metal_version(cuda_dir): function get_ext (line 43) | def get_ext(): FILE: Mamba/kernels/selective_scan/test_selective_scan.py function build_selective_scan_fn (line 18) | def build_selective_scan_fn(selective_scan_cuda: object = None, mode="ma... function selective_scan_ref (line 168) | def selective_scan_ref(u, delta, A, B, C, D=None, z=None, delta_bias=Non... function selective_scan_ref_v2 (line 237) | def selective_scan_ref_v2(u, delta, A, B, C, D=None, z=None, delta_bias=... function selective_scan_fn (line 309) | def selective_scan_fn(u, delta, A, B, C, D=None, z=None, delta_bias=None... function test_selective_scan (line 391) | def test_selective_scan(is_variable_B, is_variable_C, varBC_groups, has_... FILE: RealSR/Metric/DISTS/DISTS_pytorch/DISTS_pt.py class L2pooling (line 11) | class L2pooling(nn.Module): method __init__ (line 12) | def __init__(self, filter_size=5, stride=2, channels=None, pad_off=0): method forward (line 22) | def forward(self, input): class DISTS (line 27) | class DISTS(torch.nn.Module): method __init__ (line 28) | def __init__(self, load_weights=True): method forward_once (line 68) | def forward_once(self, x): method forward (line 82) | def forward(self, x, y, require_grad=False, batch_average=False): function prepare_image (line 115) | def prepare_image(image, resize=True): FILE: RealSR/Metric/DISTS/DISTS_tensorflow/DISTS_tf.py class DISTS (line 13) | class DISTS(): method __init__ (line 14) | def __init__(self): method get_features (line 25) | def get_features(self, img): method conv_layer (line 53) | def conv_layer(self, input, name): method pool_layer (line 61) | def pool_layer(self, input, name): method get_conv_filter (line 68) | def get_conv_filter(self, name): method get_bias (line 71) | def get_bias(self, name): method get_score (line 74) | def get_score(self, img1, img2): FILE: RealSR/Metric/LPIPS.py function main (line 11) | def main(): FILE: RealSR/Metric/PSNR.py function main (line 12) | def main(): FILE: RealSR/Metric/dists.py class L2pooling (line 13) | class L2pooling(nn.Module): method __init__ (line 14) | def __init__(self, filter_size=5, stride=2, channels=None, pad_off=0): method forward (line 24) | def forward(self, input): class DISTS (line 29) | class DISTS(torch.nn.Module): method __init__ (line 30) | def __init__(self, load_weights=True): method forward_once (line 70) | def forward_once(self, x): method forward (line 84) | def forward(self, x, y, require_grad=False, batch_average=False): function prepare_image (line 117) | def prepare_image(image, resize=True): FILE: RealSR/VmambaIR/archs/MambaRealSR11_arch.py class SelectiveScanFn (line 34) | class SelectiveScanFn(torch.autograd.Function): method forward (line 37) | def forward(ctx, u, delta, A, B, C, D=None, delta_bias=None, delta_sof... method backward (line 74) | def backward(ctx, dout, *args): function selective_scan_fn_v1 (line 101) | def selective_scan_fn_v1(u, delta, A, B, C, D=None, delta_bias=None, del... function flops_selective_scan_fn (line 110) | def flops_selective_scan_fn(B=1, L=256, D=768, N=16, with_D=True, with_Z... function print_jit_input_names (line 133) | def print_jit_input_names(inputs): function selective_scan_flop_jit (line 142) | def selective_scan_flop_jit(inputs, outputs): function to_3d (line 152) | def to_3d(x): function to_4d (line 155) | def to_4d(x,h,w): class BiasFree_LayerNorm (line 158) | class BiasFree_LayerNorm(nn.Module): method __init__ (line 159) | def __init__(self, normalized_shape): method forward (line 170) | def forward(self, x): class WithBias_LayerNorm (line 174) | class WithBias_LayerNorm(nn.Module): method __init__ (line 175) | def __init__(self, normalized_shape): method forward (line 187) | def forward(self, x): class LayerNorm (line 193) | class LayerNorm(nn.Module): method __init__ (line 194) | def __init__(self, dim, LayerNorm_type): method forward (line 201) | def forward(self, x): class FeedForward (line 209) | class FeedForward(nn.Module): method __init__ (line 210) | def __init__(self, dim, ffn_expansion_factor, bias): method forward (line 221) | def forward(self, x): class Attention (line 232) | class Attention(nn.Module): method __init__ (line 233) | def __init__(self, dim, num_heads, bias): method forward (line 244) | def forward(self, x): class SelectiveScan (line 267) | class SelectiveScan(torch.autograd.Function): method forward (line 271) | def forward(ctx, u, delta, A, B, C, D=None, delta_bias=None, delta_sof... method backward (line 304) | def backward(ctx, dout, *args): class CrossScan (line 325) | class CrossScan(torch.autograd.Function): method forward (line 327) | def forward(ctx, x: torch.Tensor): method backward (line 337) | def backward(ctx, ys: torch.Tensor): class CrossMerge (line 346) | class CrossMerge(torch.autograd.Function): method forward (line 348) | def forward(ctx, ys: torch.Tensor): method backward (line 358) | def backward(ctx, x: torch.Tensor): function cross_selective_scan (line 371) | def cross_selective_scan( class CrossScanC (line 438) | class CrossScanC(torch.autograd.Function): method forward (line 441) | def forward(ctx, x: torch.Tensor): method backward (line 451) | def backward(ctx, ys: torch.Tensor): class CrossMergeC (line 459) | class CrossMergeC(torch.autograd.Function): method forward (line 462) | def forward(ctx, ys: torch.Tensor): method backward (line 469) | def backward(ctx, x: torch.Tensor): function cross_selective_scanC (line 478) | def cross_selective_scanC( class SS2D_1 (line 540) | class SS2D_1(nn.Module): method __init__ (line 541) | def __init__( method dt_init (line 670) | def dt_init(dt_rank, d_inner, dt_scale=1.0, dt_init="random", dt_min=0... method A_log_init (line 695) | def A_log_init(d_state, d_inner, copies=-1, device=None, merge=True): method D_init (line 712) | def D_init(d_inner, copies=-1, device=None, merge=True): method forward_corev1 (line 723) | def forward_corev1(self, x: torch.Tensor): method forward_corev2 (line 791) | def forward_corev2(self, x: torch.Tensor, nrows=-1, channel_first=False): method cforward_corev2 (line 806) | def cforward_corev2(self, xc: torch.Tensor, nrows=-1, channel_first=Fa... method forward (line 819) | def forward(self, x: torch.Tensor, **kwargs): class MamberBlock (line 835) | class MamberBlock(nn.Module): method __init__ (line 836) | def __init__(self, dim, num_heads, ffn_expansion_factor, bias, LayerNo... method forward (line 844) | def forward(self, x): class OverlapPatchEmbed (line 854) | class OverlapPatchEmbed(nn.Module): method __init__ (line 855) | def __init__(self, in_c=3, embed_dim=48, bias=False): method forward (line 860) | def forward(self, x): class Downsample (line 869) | class Downsample(nn.Module): method __init__ (line 870) | def __init__(self, n_feat): method forward (line 876) | def forward(self, x): class Upsample (line 879) | class Upsample(nn.Module): method __init__ (line 880) | def __init__(self, n_feat): method forward (line 886) | def forward(self, x): class MambaRealSR11 (line 892) | class MambaRealSR11(nn.Module): method __init__ (line 893) | def __init__(self, method forward (line 944) | def forward(self, inp_img): method flops (line 980) | def flops(self, shape=(3, 64, 64)): FILE: RealSR/VmambaIR/archs/attention.py class QKVAttentionLegacy (line 4) | class QKVAttentionLegacy(nn.Module): method __init__ (line 9) | def __init__(self, n_heads): method forward (line 14) | def forward(self, qkv): class QKVAttention (line 32) | class QKVAttention(nn.Module): method __init__ (line 37) | def __init__(self, n_heads): method forward (line 42) | def forward(self, qkv): class AttentionBlock (line 62) | class AttentionBlock(nn.Module): method __init__ (line 63) | def __init__( method forward (line 88) | def forward(self, x): FILE: RealSR/VmambaIR/archs/common.py function default_conv (line 7) | def default_conv(in_channels, out_channels, kernel_size, bias=True): class ResBlock (line 10) | class ResBlock(nn.Module): method __init__ (line 11) | def __init__( method forward (line 27) | def forward(self, x): class MeanShift (line 33) | class MeanShift(nn.Conv2d): method __init__ (line 34) | def __init__(self, rgb_range, rgb_mean, rgb_std, sign=-1): class Upsampler (line 45) | class Upsampler(nn.Sequential): method __init__ (line 46) | def __init__(self, conv, scale, n_feat, act=False, bias=True): FILE: RealSR/VmambaIR/archs/discriminator_arch.py class UNetDiscriminatorSN (line 8) | class UNetDiscriminatorSN(nn.Module): method __init__ (line 19) | def __init__(self, num_in_ch, num_feat=64, skip_connection=True): method forward (line 38) | def forward(self, x): FILE: RealSR/VmambaIR/archs/srvgg_arch.py class SRVGGNetCompact (line 7) | class SRVGGNetCompact(nn.Module): method __init__ (line 22) | def __init__(self, num_in_ch=3, num_out_ch=3, num_feat=64, num_conv=16... method forward (line 60) | def forward(self, x): FILE: RealSR/VmambaIR/data/data_util.py function read_img_seq (line 12) | def read_img_seq(path, require_mod_crop=False, scale=1): function generate_frame_indices (line 36) | def generate_frame_indices(crt_idx, function paired_paths_from_lmdb (line 92) | def paired_paths_from_lmdb(folders, keys): function paired_paths_from_meta_info_file (line 158) | def paired_paths_from_meta_info_file(folders, keys, meta_info_file, function paired_paths_from_folder (line 208) | def paired_paths_from_folder(folders, keys, filename_tmpl): function paired_DP_paths_from_folder (line 253) | def paired_DP_paths_from_folder(folders, keys, filename_tmpl): function paths_from_folder (line 307) | def paths_from_folder(folder): function paths_from_lmdb (line 322) | def paths_from_lmdb(folder): function generate_gaussian_kernel (line 338) | def generate_gaussian_kernel(kernel_size=13, sigma=1.6): function duf_downsample (line 356) | def duf_downsample(x, kernel_size=13, scale=4): FILE: RealSR/VmambaIR/data/deblur_paired_dataset.py class DeblurPairedDataset (line 17) | class DeblurPairedDataset(data.Dataset): method __init__ (line 46) | def __init__(self, opt): method __getitem__ (line 79) | def __getitem__(self, index): method __len__ (line 132) | def __len__(self): class Dataset_GaussianDenoising (line 135) | class Dataset_GaussianDenoising(data.Dataset): method __init__ (line 163) | def __init__(self, opt): method __getitem__ (line 199) | def __getitem__(self, index): method __len__ (line 274) | def __len__(self): class Dataset_DefocusDeblur_DualPixel_16bit (line 277) | class Dataset_DefocusDeblur_DualPixel_16bit(data.Dataset): method __init__ (line 278) | def __init__(self, opt): method __getitem__ (line 300) | def __getitem__(self, index): method __len__ (line 363) | def __len__(self): FILE: RealSR/VmambaIR/data/diffir_dataset.py class DiffIRGANDataset (line 17) | class DiffIRGANDataset(data.Dataset): method __init__ (line 35) | def __init__(self, opt): method __getitem__ (line 57) | def __getitem__(self, index): method __len__ (line 108) | def __len__(self): FILE: RealSR/VmambaIR/data/diffir_paired_dataset.py class DiffIRGANPairedDataset (line 11) | class DiffIRGANPairedDataset(data.Dataset): method __init__ (line 41) | def __init__(self, opt): method __getitem__ (line 75) | def __getitem__(self, index): method __len__ (line 107) | def __len__(self): FILE: RealSR/VmambaIR/data/gaussiandenoising_paired_dataset.py class GaussianDenoisingPairedDataset (line 18) | class GaussianDenoisingPairedDataset(data.Dataset): method __init__ (line 46) | def __init__(self, opt): method __getitem__ (line 82) | def __getitem__(self, index): method __len__ (line 158) | def __len__(self): class Dataset_DefocusDeblur_DualPixel_16bit (line 161) | class Dataset_DefocusDeblur_DualPixel_16bit(data.Dataset): method __init__ (line 162) | def __init__(self, opt): method __getitem__ (line 184) | def __getitem__(self, index): method __len__ (line 247) | def __len__(self): FILE: RealSR/VmambaIR/data/realesrgan400_dataset.py class RealESRGANDataset400 (line 17) | class RealESRGANDataset400(data.Dataset): method __init__ (line 35) | def __init__(self, opt): method __getitem__ (line 83) | def __getitem__(self, index): method __len__ (line 191) | def __len__(self): FILE: RealSR/VmambaIR/data/realesrgan_dataset.py class RealESRGANDataset (line 17) | class RealESRGANDataset(data.Dataset): method __init__ (line 35) | def __init__(self, opt): method __getitem__ (line 83) | def __getitem__(self, index): method __len__ (line 191) | def __len__(self): FILE: RealSR/VmambaIR/data/realesrgan_memery_dataset.py class RealESRGANDataset_memory (line 17) | class RealESRGANDataset_memory(data.Dataset): method __init__ (line 35) | def __init__(self, opt): method __getitem__ (line 83) | def __getitem__(self, index): method __len__ (line 191) | def __len__(self): FILE: RealSR/VmambaIR/data/realesrgan_paired_dataset.py class RealESRGANPairedDataset (line 11) | class RealESRGANPairedDataset(data.Dataset): method __init__ (line 41) | def __init__(self, opt): method __getitem__ (line 75) | def __getitem__(self, index): method __len__ (line 107) | def __len__(self): FILE: RealSR/VmambaIR/data/transforms.py function mod_crop (line 5) | def mod_crop(img, scale): function paired_random_crop (line 24) | def paired_random_crop(img_gts, img_lqs, lq_patch_size, scale, gt_path): function paired_random_crop_DP (line 85) | def paired_random_crop_DP(img_lqLs, img_lqRs, img_gts, gt_patch_size, sc... function augment (line 136) | def augment(imgs, hflip=True, rotation=True, flows=None, return_status=F... function img_rotate (line 203) | def img_rotate(img, angle, center=None, scale=1.0): function data_augmentation (line 223) | def data_augmentation(image, mode): function random_augmentation (line 270) | def random_augmentation(*args): FILE: RealSR/VmambaIR/losses/my_loss.py class KDLoss (line 8) | class KDLoss(nn.Module): method __init__ (line 14) | def __init__(self, loss_weight=1.0, temperature = 0.15): method forward (line 20) | def forward(self, S1_fea, S2_fea): FILE: RealSR/VmambaIR/models/MambaRealSRGAN_model.py class MambaRealSRGAN (line 18) | class MambaRealSRGAN(SRGANModel): method __init__ (line 26) | def __init__(self, opt): method setup_optimizers (line 56) | def setup_optimizers(self): method init_training_settings (line 83) | def init_training_settings(self): method _dequeue_and_enqueue (line 89) | def _dequeue_and_enqueue(self): method feed_data (line 126) | def feed_data(self, data): method nondist_validation (line 247) | def nondist_validation(self, dataloader, current_iter, tb_logger, save... method pad_test (line 253) | def pad_test(self, window_size): method test (line 265) | def test(self): method optimize_parameters (line 289) | def optimize_parameters(self, current_iter): FILE: RealSR/VmambaIR/models/MambaRealSRGANtest_model.py class MambaRealSRGANtest (line 18) | class MambaRealSRGANtest(SRGANModel): method __init__ (line 26) | def __init__(self, opt): method setup_optimizers (line 56) | def setup_optimizers(self): method init_training_settings (line 83) | def init_training_settings(self): method _dequeue_and_enqueue (line 89) | def _dequeue_and_enqueue(self): method feed_data (line 126) | def feed_data(self, data): method nondist_validation (line 247) | def nondist_validation(self, dataloader, current_iter, tb_logger, save... method pad_test (line 253) | def pad_test(self, window_size): method test (line 265) | def test(self): method optimize_parameters (line 289) | def optimize_parameters(self, current_iter): FILE: RealSR/VmambaIR/models/MambaRealSR_model.py class Mixing_Augment (line 14) | class Mixing_Augment: method __init__ (line 15) | def __init__(self, mixup_beta, use_identity, device): method mixup (line 23) | def mixup(self, target, input_): method __call__ (line 33) | def __call__(self, target, input_): class MambaRealSR (line 44) | class MambaRealSR(SRModel): method __init__ (line 52) | def __init__(self, opt): method setup_schedulers (line 60) | def setup_schedulers(self): method _dequeue_and_enqueue (line 110) | def _dequeue_and_enqueue(self): method feed_data (line 147) | def feed_data(self, data): method nondist_validation (line 266) | def nondist_validation(self, dataloader, current_iter, tb_logger, save... method pad_test (line 272) | def pad_test(self, window_size): method test (line 284) | def test(self): method optimize_parameters (line 305) | def optimize_parameters(self, current_iter): FILE: RealSR/VmambaIR/models/lr_scheduler.py class MultiStepRestartLR (line 7) | class MultiStepRestartLR(_LRScheduler): method __init__ (line 20) | def __init__(self, method get_lr (line 35) | def get_lr(self): class LinearLR (line 49) | class LinearLR(_LRScheduler): method __init__ (line 59) | def __init__(self, method get_lr (line 66) | def get_lr(self): class VibrateLR (line 72) | class VibrateLR(_LRScheduler): method __init__ (line 82) | def __init__(self, method get_lr (line 89) | def get_lr(self): function get_position_from_periods (line 115) | def get_position_from_periods(iteration, cumulative_period): class CosineAnnealingRestartLR (line 136) | class CosineAnnealingRestartLR(_LRScheduler): method __init__ (line 156) | def __init__(self, method get_lr (line 172) | def get_lr(self): class CosineAnnealingRestartCyclicLR (line 186) | class CosineAnnealingRestartCyclicLR(_LRScheduler): method __init__ (line 203) | def __init__(self, method get_lr (line 219) | def get_lr(self): FILE: RealSR/VmambaIR/train_pipeline.py function init_tb_loggers (line 23) | def init_tb_loggers(opt): function create_train_val_dataloader (line 35) | def create_train_val_dataloader(opt, logger): function load_resume_state (line 74) | def load_resume_state(opt): function train_pipeline (line 97) | def train_pipeline(root_path): FILE: RealSR/VmambaIR/utils.py class RealESRGANer (line 14) | class RealESRGANer(): method __init__ (line 29) | def __init__(self, method pre_process (line 68) | def pre_process(self, img): method process (line 93) | def process(self): method tile_process (line 97) | def tile_process(self): method post_process (line 162) | def post_process(self): method enhance (line 174) | def enhance(self, img, outscale=None, alpha_upsampler='realesrgan'): class PrefetchReader (line 246) | class PrefetchReader(threading.Thread): method __init__ (line 254) | def __init__(self, img_list, num_prefetch_queue): method run (line 259) | def run(self): method __next__ (line 266) | def __next__(self): method __iter__ (line 272) | def __iter__(self): class IOConsumer (line 276) | class IOConsumer(threading.Thread): method __init__ (line 278) | def __init__(self, opt, que, qid): method run (line 284) | def run(self): FILE: RealSR/VmambaIR/utils/bundle_submissions.py function bundle_submissions_raw (line 13) | def bundle_submissions_raw(submission_folder,session): function bundle_submissions_srgb (line 46) | def bundle_submissions_srgb(submission_folder,session): function bundle_submissions_srgb_v1 (line 79) | def bundle_submissions_srgb_v1(submission_folder,session): FILE: RealSR/VmambaIR/utils/create_lmdb.py function prepare_keys (line 7) | def prepare_keys(folder_path, suffix='png'): function create_lmdb_for_reds (line 24) | def create_lmdb_for_reds(): function create_lmdb_for_gopro (line 46) | def create_lmdb_for_gopro(): function create_lmdb_for_rain13k (line 71) | def create_lmdb_for_rain13k(): function create_lmdb_for_SIDD (line 84) | def create_lmdb_for_SIDD(): FILE: RealSR/VmambaIR/utils/dist_util.py function init_dist (line 10) | def init_dist(launcher, backend='nccl', **kwargs): function _init_dist_pytorch (line 21) | def _init_dist_pytorch(backend, **kwargs): function _init_dist_slurm (line 28) | def _init_dist_slurm(backend, port=None): function get_dist_info (line 61) | def get_dist_info(): function master_only (line 75) | def master_only(func): FILE: RealSR/VmambaIR/utils/download_util.py function download_file_from_google_drive (line 8) | def download_file_from_google_drive(file_id, save_path): function get_confirm_token (line 41) | def get_confirm_token(response): function save_response_content (line 48) | def save_response_content(response, FILE: RealSR/VmambaIR/utils/face_util.py class FaceRestorationHelper (line 16) | class FaceRestorationHelper(object): method __init__ (line 19) | def __init__(self, upscale_factor, face_size=512): method init_dlib (line 41) | def init_dlib(self, detection_path, landmark5_path, landmark68_path): method free_dlib_gpu_memory (line 47) | def free_dlib_gpu_memory(self): method read_input_image (line 52) | def read_input_image(self, img_path): method detect_faces (line 56) | def detect_faces(self, method get_face_landmarks_5 (line 89) | def get_face_landmarks_5(self): method get_face_landmarks_68 (line 96) | def get_face_landmarks_68(self): method warp_crop_faces (line 132) | def warp_crop_faces(self, method add_restored_face (line 170) | def add_restored_face(self, face): method paste_faces_to_input_image (line 173) | def paste_faces_to_input_image(self, save_path): method clean_all (line 211) | def clean_all(self): FILE: RealSR/VmambaIR/utils/file_client.py class BaseStorageBackend (line 5) | class BaseStorageBackend(metaclass=ABCMeta): method get (line 14) | def get(self, filepath): method get_text (line 18) | def get_text(self, filepath): class MemcachedBackend (line 22) | class MemcachedBackend(BaseStorageBackend): method __init__ (line 32) | def __init__(self, server_list_cfg, client_cfg, sys_path=None): method get (line 49) | def get(self, filepath): method get_text (line 56) | def get_text(self, filepath): class HardDiskBackend (line 60) | class HardDiskBackend(BaseStorageBackend): method get (line 63) | def get(self, filepath): method get_text (line 69) | def get_text(self, filepath): class LmdbBackend (line 76) | class LmdbBackend(BaseStorageBackend): method __init__ (line 96) | def __init__(self, method get (line 131) | def get(self, filepath, client_key): method get_text (line 146) | def get_text(self, filepath): class FileClient (line 150) | class FileClient(object): method __init__ (line 169) | def __init__(self, backend='disk', **kwargs): method get (line 177) | def get(self, filepath, client_key='default'): method get_text (line 185) | def get_text(self, filepath): FILE: RealSR/VmambaIR/utils/flow_util.py function flowread (line 7) | def flowread(flow_path, quantize=False, concat_axis=0, *args, **kwargs): function flowwrite (line 47) | def flowwrite(flow, filename, quantize=False, concat_axis=0, *args, **kw... function quantize_flow (line 78) | def quantize_flow(flow, max_val=0.02, norm=True): function dequantize_flow (line 106) | def dequantize_flow(dx, dy, max_val=0.02, denorm=True): function quantize (line 130) | def quantize(arr, min_val, max_val, levels, dtype=np.int64): function dequantize (line 157) | def dequantize(arr, min_val, max_val, levels, dtype=np.float64): FILE: RealSR/VmambaIR/utils/img_util.py function img2tensor (line 9) | def img2tensor(imgs, bgr2rgb=True, float32=True): function tensor2img (line 36) | def tensor2img(tensor, rgb2bgr=True, out_type=np.uint8, min_max=(0, 1)): function imfrombytes (line 101) | def imfrombytes(content, flag='color', float32=False): function imfrombytesDP (line 127) | def imfrombytesDP(content, flag='color', float32=False): function padding (line 148) | def padding(img_lq, img_gt, gt_size): function padding_DP (line 166) | def padding_DP(img_lqL, img_lqR, img_gt, gt_size): function imwrite (line 181) | def imwrite(img, file_path, params=None, auto_mkdir=True): function crop_border (line 200) | def crop_border(imgs, crop_border): FILE: RealSR/VmambaIR/utils/lmdb_util.py function make_lmdb_from_imgs (line 9) | def make_lmdb_from_imgs(data_path, function read_img_worker (line 139) | def read_img_worker(path, key, compress_level): class LmdbMaker (line 164) | class LmdbMaker(): method __init__ (line 175) | def __init__(self, method put (line 194) | def put(self, img_byte, key, img_shape): method close (line 205) | def close(self): FILE: RealSR/VmambaIR/utils/logger.py class MessageLogger (line 10) | class MessageLogger(): method __init__ (line 23) | def __init__(self, opt, start_iter=1, tb_logger=None): method __call__ (line 34) | def __call__(self, log_vars): function init_tb_logger (line 81) | def init_tb_logger(log_dir): function init_wandb_logger (line 88) | def init_wandb_logger(opt): function get_root_logger (line 108) | def get_root_logger(logger_name='basicsr', log_level=logging.INFO, log_f... function get_env_info (line 150) | def get_env_info(): FILE: RealSR/VmambaIR/utils/matlab_functions.py function cubic (line 6) | def cubic(x): function calculate_weights_indices (line 17) | def calculate_weights_indices(in_length, out_length, scale, kernel, function imresize (line 88) | def imresize(img, scale, antialiasing=True): function rgb2ycbcr (line 173) | def rgb2ycbcr(img, y_only=False): function bgr2ycbcr (line 207) | def bgr2ycbcr(img, y_only=False): function ycbcr2rgb (line 241) | def ycbcr2rgb(img): function ycbcr2bgr (line 273) | def ycbcr2bgr(img): function _convert_input_type_range (line 305) | def _convert_input_type_range(img): function _convert_output_type_range (line 333) | def _convert_output_type_range(img, dst_type): FILE: RealSR/VmambaIR/utils/misc.py function set_random_seed (line 12) | def set_random_seed(seed): function get_time_str (line 21) | def get_time_str(): function mkdir_and_rename (line 25) | def mkdir_and_rename(path): function make_exp_dirs (line 39) | def make_exp_dirs(opt): function scandir (line 53) | def scandir(dir_path, suffix=None, recursive=False, full_path=False): function scandir_SIDD (line 95) | def scandir_SIDD(dir_path, keywords=None, recursive=False, full_path=Fal... function check_resume (line 137) | def check_resume(opt, resume_iter): function sizeof_fmt (line 166) | def sizeof_fmt(size, suffix='B'): FILE: RealSR/VmambaIR/utils/options.py function ordered_yaml (line 6) | def ordered_yaml(): function parse (line 31) | def parse(opt_path, is_train=True): function dict2str (line 92) | def dict2str(opt, indent_level=1): FILE: RealSR/inference.py function pad_test (line 19) | def pad_test(lq,scale): FILE: RealSR/ldm/classifier.py function disabled_train (line 22) | def disabled_train(self, mode=True): class NoisyLatentImageClassifier (line 28) | class NoisyLatentImageClassifier(pl.LightningModule): method __init__ (line 30) | def __init__(self, method init_from_ckpt (line 70) | def init_from_ckpt(self, path, ignore_keys=list(), only_model=False): method load_diffusion (line 88) | def load_diffusion(self): method load_classifier (line 95) | def load_classifier(self, ckpt_path, pool): method get_x_noisy (line 110) | def get_x_noisy(self, x, t, noise=None): method forward (line 120) | def forward(self, x_noisy, t, *args, **kwargs): method get_input (line 124) | def get_input(self, batch, k): method get_conditioning (line 133) | def get_conditioning(self, batch, k=None): method compute_top_k (line 150) | def compute_top_k(self, logits, labels, k, reduction="mean"): method on_train_epoch_start (line 157) | def on_train_epoch_start(self): method write_logs (line 162) | def write_logs(self, loss, logits, targets): method shared_step (line 179) | def shared_step(self, batch, t=None): method training_step (line 198) | def training_step(self, batch, batch_idx): method reset_noise_accs (line 202) | def reset_noise_accs(self): method on_validation_start (line 206) | def on_validation_start(self): method validation_step (line 210) | def validation_step(self, batch, batch_idx): method configure_optimizers (line 220) | def configure_optimizers(self): method log_images (line 238) | def log_images(self, batch, N=8, *args, **kwargs): FILE: RealSR/ldm/lr_scheduler.py class LambdaWarmUpCosineScheduler (line 4) | class LambdaWarmUpCosineScheduler: method __init__ (line 8) | def __init__(self, warm_up_steps, lr_min, lr_max, lr_start, max_decay_... method schedule (line 17) | def schedule(self, n, **kwargs): method __call__ (line 32) | def __call__(self, n, **kwargs): class LambdaWarmUpCosineScheduler2 (line 36) | class LambdaWarmUpCosineScheduler2: method __init__ (line 41) | def __init__(self, warm_up_steps, f_min, f_max, f_start, cycle_lengths... method find_in_interval (line 52) | def find_in_interval(self, n): method schedule (line 59) | def schedule(self, n, **kwargs): method __call__ (line 77) | def __call__(self, n, **kwargs): class LambdaLinearScheduler (line 81) | class LambdaLinearScheduler(LambdaWarmUpCosineScheduler2): method schedule (line 83) | def schedule(self, n, **kwargs): FILE: RealSR/ldm/util.py function log_txt_as_img (line 17) | def log_txt_as_img(wh, xc, size=10): function ismap (line 41) | def ismap(x): function isimage (line 47) | def isimage(x): function exists (line 53) | def exists(x): function default (line 57) | def default(val, d): function mean_flat (line 63) | def mean_flat(tensor): function count_params (line 71) | def count_params(model, verbose=False): function instantiate_from_config (line 78) | def instantiate_from_config(config): function get_obj_from_str (line 88) | def get_obj_from_str(string, reload=False): function _do_parallel_data_prefetch (line 96) | def _do_parallel_data_prefetch(func, Q, data, idx, idx_to_fn=False): function parallel_data_prefetch (line 108) | def parallel_data_prefetch( FILE: RealSR/ldm/util2.py function make_beta_schedule (line 21) | def make_beta_schedule(schedule, n_timestep, linear_start=1e-4, linear_e... function make_ddim_timesteps (line 46) | def make_ddim_timesteps(ddim_discr_method, num_ddim_timesteps, num_ddpm_... function make_ddim_sampling_parameters (line 63) | def make_ddim_sampling_parameters(alphacums, ddim_timesteps, eta, verbos... function betas_for_alpha_bar (line 77) | def betas_for_alpha_bar(num_diffusion_timesteps, alpha_bar, max_beta=0.9... function extract_into_tensor (line 96) | def extract_into_tensor(a, t, x_shape): function checkpoint (line 102) | def checkpoint(func, inputs, params, flag): class CheckpointFunction (line 119) | class CheckpointFunction(torch.autograd.Function): method forward (line 121) | def forward(ctx, run_function, length, *args): method backward (line 131) | def backward(ctx, *output_grads): function timestep_embedding (line 151) | def timestep_embedding(timesteps, dim, max_period=10000, repeat_only=Fal... function zero_module (line 174) | def zero_module(module): function scale_module (line 183) | def scale_module(module, scale): function mean_flat (line 192) | def mean_flat(tensor): function normalization (line 199) | def normalization(channels): class SiLU (line 209) | class SiLU(nn.Module): method forward (line 210) | def forward(self, x): class GroupNorm32 (line 214) | class GroupNorm32(nn.GroupNorm): method forward (line 215) | def forward(self, x): function conv_nd (line 218) | def conv_nd(dims, *args, **kwargs): function linear (line 231) | def linear(*args, **kwargs): function avg_pool_nd (line 238) | def avg_pool_nd(dims, *args, **kwargs): class HybridConditioner (line 251) | class HybridConditioner(nn.Module): method __init__ (line 253) | def __init__(self, c_concat_config, c_crossattn_config): method forward (line 258) | def forward(self, c_concat, c_crossattn): function noise_like (line 264) | def noise_like(shape, device, repeat=False): FILE: RealSR/scripts/Metric/DISTS/DISTS_pytorch/DISTS_pt.py class L2pooling (line 11) | class L2pooling(nn.Module): method __init__ (line 12) | def __init__(self, filter_size=5, stride=2, channels=None, pad_off=0): method forward (line 22) | def forward(self, input): class DISTS (line 27) | class DISTS(torch.nn.Module): method __init__ (line 28) | def __init__(self, load_weights=True): method forward_once (line 68) | def forward_once(self, x): method forward (line 82) | def forward(self, x, y, require_grad=False, batch_average=False): function prepare_image (line 115) | def prepare_image(image, resize=True): FILE: RealSR/scripts/Metric/DISTS/DISTS_tensorflow/DISTS_tf.py class DISTS (line 13) | class DISTS(): method __init__ (line 14) | def __init__(self): method get_features (line 25) | def get_features(self, img): method conv_layer (line 53) | def conv_layer(self, input, name): method pool_layer (line 61) | def pool_layer(self, input, name): method get_conv_filter (line 68) | def get_conv_filter(self, name): method get_bias (line 71) | def get_bias(self, name): method get_score (line 74) | def get_score(self, img1, img2): FILE: RealSR/scripts/Metric/LPIPS.py function main (line 11) | def main(): FILE: RealSR/scripts/Metric/PSNR.py function main (line 12) | def main(): FILE: RealSR/scripts/Metric/dists.py class L2pooling (line 13) | class L2pooling(nn.Module): method __init__ (line 14) | def __init__(self, filter_size=5, stride=2, channels=None, pad_off=0): method forward (line 24) | def forward(self, input): class DISTS (line 29) | class DISTS(torch.nn.Module): method __init__ (line 30) | def __init__(self, load_weights=True): method forward_once (line 70) | def forward_once(self, x): method forward (line 84) | def forward(self, x, y, require_grad=False, batch_average=False): function prepare_image (line 117) | def prepare_image(image, resize=True): FILE: RealSR/scripts/extract_subimages.py function main (line 12) | def main(args): function extract_subimages (line 43) | def extract_subimages(opt): function worker (line 74) | def worker(path, opt): FILE: RealSR/scripts/extract_subimages_DF2K.py function main (line 12) | def main(args): function extract_subimages (line 43) | def extract_subimages(opt): function worker (line 74) | def worker(path, opt): FILE: RealSR/scripts/generate_meta_info.py function main (line 7) | def main(args): FILE: RealSR/scripts/generate_meta_info_DF2K.py function main (line 7) | def main(args): FILE: RealSR/scripts/generate_meta_info_DF2K_memo.py function main (line 7) | def main(args): FILE: RealSR/scripts/generate_meta_info_OST.py function main (line 7) | def main(args): FILE: RealSR/scripts/generate_meta_info_pairdata.py function main (line 6) | def main(args): FILE: RealSR/scripts/generate_multiscale_DF2K.py function main (line 7) | def main(args): FILE: RealSR/scripts/pytorch2onnx.py function main (line 7) | def main(args): FILE: RealSR/setup.py function readme (line 12) | def readme(): function get_git_hash (line 18) | def get_git_hash(): function get_hash (line 43) | def get_hash(): function write_version_py (line 52) | def write_version_py(): function get_version (line 69) | def get_version(): function get_requirements (line 75) | def get_requirements(filename='requirements.txt'): FILE: RealSR/tests/test_dataset.py function test_realesrgan_dataset (line 8) | def test_realesrgan_dataset(): function test_realesrgan_paired_dataset (line 82) | def test_realesrgan_paired_dataset(): FILE: RealSR/tests/test_discriminator_arch.py function test_unetdiscriminatorsn (line 6) | def test_unetdiscriminatorsn(): FILE: RealSR/tests/test_model.py function test_realesrnet_model (line 12) | def test_realesrnet_model(): function test_realesrgan_model (line 64) | def test_realesrgan_model(): FILE: RealSR/tests/test_utils.py function test_realesrganer (line 7) | def test_realesrganer(): FILE: SRGAN/Metric/DISTS/DISTS_pytorch/DISTS_pt.py class L2pooling (line 11) | class L2pooling(nn.Module): method __init__ (line 12) | def __init__(self, filter_size=5, stride=2, channels=None, pad_off=0): method forward (line 22) | def forward(self, input): class DISTS (line 27) | class DISTS(torch.nn.Module): method __init__ (line 28) | def __init__(self, load_weights=True): method forward_once (line 68) | def forward_once(self, x): method forward (line 82) | def forward(self, x, y, require_grad=False, batch_average=False): function prepare_image (line 115) | def prepare_image(image, resize=True): FILE: SRGAN/Metric/DISTS/DISTS_tensorflow/DISTS_tf.py class DISTS (line 13) | class DISTS(): method __init__ (line 14) | def __init__(self): method get_features (line 25) | def get_features(self, img): method conv_layer (line 53) | def conv_layer(self, input, name): method pool_layer (line 61) | def pool_layer(self, input, name): method get_conv_filter (line 68) | def get_conv_filter(self, name): method get_bias (line 71) | def get_bias(self, name): method get_score (line 74) | def get_score(self, img1, img2): FILE: SRGAN/Metric/LPIPS.py function main (line 11) | def main(): FILE: SRGAN/Metric/PSNR.py function main (line 12) | def main(): FILE: SRGAN/Metric/dists.py class L2pooling (line 13) | class L2pooling(nn.Module): method __init__ (line 14) | def __init__(self, filter_size=5, stride=2, channels=None, pad_off=0): method forward (line 24) | def forward(self, input): class DISTS (line 29) | class DISTS(torch.nn.Module): method __init__ (line 30) | def __init__(self, load_weights=True): method forward_once (line 70) | def forward_once(self, x): method forward (line 84) | def forward(self, x, y, require_grad=False, batch_average=False): function prepare_image (line 117) | def prepare_image(image, resize=True): FILE: SRGAN/VmambaIR/archs/MambaSISR6_arch.py class SelectiveScanFn (line 24) | class SelectiveScanFn(torch.autograd.Function): method forward (line 27) | def forward(ctx, u, delta, A, B, C, D=None, delta_bias=None, delta_sof... method backward (line 64) | def backward(ctx, dout, *args): function selective_scan_fn_v1 (line 91) | def selective_scan_fn_v1(u, delta, A, B, C, D=None, delta_bias=None, del... function flops_selective_scan_fn (line 101) | def flops_selective_scan_fn(B=1, L=256, D=768, N=16, with_D=True, with_Z... function print_jit_input_names (line 124) | def print_jit_input_names(inputs): function selective_scan_flop_jit (line 133) | def selective_scan_flop_jit(inputs, outputs): function to_3d (line 144) | def to_3d(x): function to_4d (line 147) | def to_4d(x,h,w): class BiasFree_LayerNorm (line 150) | class BiasFree_LayerNorm(nn.Module): method __init__ (line 151) | def __init__(self, normalized_shape): method forward (line 162) | def forward(self, x): class WithBias_LayerNorm (line 166) | class WithBias_LayerNorm(nn.Module): method __init__ (line 167) | def __init__(self, normalized_shape): method forward (line 179) | def forward(self, x): class LayerNorm (line 185) | class LayerNorm(nn.Module): method __init__ (line 186) | def __init__(self, dim, LayerNorm_type): method forward (line 193) | def forward(self, x): class FeedForward (line 201) | class FeedForward(nn.Module): method __init__ (line 202) | def __init__(self, dim, ffn_expansion_factor, bias): method forward (line 213) | def forward(self, x): class SS2D_1 (line 222) | class SS2D_1(nn.Module): method __init__ (line 223) | def __init__( method dt_init (line 338) | def dt_init(dt_rank, d_inner, dt_scale=1.0, dt_init="random", dt_min=0... method A_log_init (line 364) | def A_log_init(d_state, d_inner, copies=-1, device=None, merge=True): method D_init (line 382) | def D_init(d_inner, copies=-1, device=None, merge=True): method forward_corev1 (line 395) | def forward_corev1(self, x: torch.Tensor): method cforward_corev1 (line 438) | def cforward_corev1(self, xc: torch.Tensor): method forward (line 486) | def forward(self, x: torch.Tensor, **kwargs): class MamberBlock (line 502) | class MamberBlock(nn.Module): method __init__ (line 503) | def __init__(self, dim, num_heads, ffn_expansion_factor, bias, LayerNo... method forward (line 511) | def forward(self, x): class OverlapPatchEmbed (line 520) | class OverlapPatchEmbed(nn.Module): method __init__ (line 521) | def __init__(self, in_c=3, embed_dim=48, bias=False): method forward (line 526) | def forward(self, x): class Downsample (line 535) | class Downsample(nn.Module): method __init__ (line 536) | def __init__(self, n_feat): method forward (line 542) | def forward(self, x): class Upsample (line 545) | class Upsample(nn.Module): method __init__ (line 546) | def __init__(self, n_feat): method forward (line 552) | def forward(self, x): class MambaSISR6 (line 558) | class MambaSISR6(nn.Module): method __init__ (line 559) | def __init__(self, method forward (line 610) | def forward(self, inp_img): method flops (line 646) | def flops(self, shape=(3, 64, 64)): FILE: SRGAN/VmambaIR/archs/attention.py class QKVAttentionLegacy (line 4) | class QKVAttentionLegacy(nn.Module): method __init__ (line 9) | def __init__(self, n_heads): method forward (line 14) | def forward(self, qkv): class QKVAttention (line 32) | class QKVAttention(nn.Module): method __init__ (line 37) | def __init__(self, n_heads): method forward (line 42) | def forward(self, qkv): class AttentionBlock (line 62) | class AttentionBlock(nn.Module): method __init__ (line 63) | def __init__( method forward (line 88) | def forward(self, x): FILE: SRGAN/VmambaIR/archs/common.py function default_conv (line 7) | def default_conv(in_channels, out_channels, kernel_size, bias=True): class ResBlock (line 10) | class ResBlock(nn.Module): method __init__ (line 11) | def __init__( method forward (line 27) | def forward(self, x): class MeanShift (line 33) | class MeanShift(nn.Conv2d): method __init__ (line 34) | def __init__(self, rgb_range, rgb_mean, rgb_std, sign=-1): class Upsampler (line 45) | class Upsampler(nn.Sequential): method __init__ (line 46) | def __init__(self, conv, scale, n_feat, act=False, bias=True): FILE: SRGAN/VmambaIR/archs/discriminator_arch.py class UNetDiscriminatorSN (line 8) | class UNetDiscriminatorSN(nn.Module): method __init__ (line 19) | def __init__(self, num_in_ch, num_feat=64, skip_connection=True): method forward (line 38) | def forward(self, x): FILE: SRGAN/VmambaIR/archs/srvgg_arch.py class SRVGGNetCompact (line 7) | class SRVGGNetCompact(nn.Module): method __init__ (line 22) | def __init__(self, num_in_ch=3, num_out_ch=3, num_feat=64, num_conv=16... method forward (line 60) | def forward(self, x): FILE: SRGAN/VmambaIR/data/data_util.py function read_img_seq (line 12) | def read_img_seq(path, require_mod_crop=False, scale=1): function generate_frame_indices (line 36) | def generate_frame_indices(crt_idx, function paired_paths_from_lmdb (line 92) | def paired_paths_from_lmdb(folders, keys): function paired_paths_from_meta_info_file (line 158) | def paired_paths_from_meta_info_file(folders, keys, meta_info_file, function paired_paths_from_folder (line 208) | def paired_paths_from_folder(folders, keys, filename_tmpl): function paired_DP_paths_from_folder (line 253) | def paired_DP_paths_from_folder(folders, keys, filename_tmpl): function paths_from_folder (line 307) | def paths_from_folder(folder): function paths_from_lmdb (line 322) | def paths_from_lmdb(folder): function generate_gaussian_kernel (line 338) | def generate_gaussian_kernel(kernel_size=13, sigma=1.6): function duf_downsample (line 356) | def duf_downsample(x, kernel_size=13, scale=4): FILE: SRGAN/VmambaIR/data/deblur_paired_dataset.py class DeblurPairedDataset (line 17) | class DeblurPairedDataset(data.Dataset): method __init__ (line 46) | def __init__(self, opt): method __getitem__ (line 79) | def __getitem__(self, index): method __len__ (line 132) | def __len__(self): class Dataset_GaussianDenoising (line 135) | class Dataset_GaussianDenoising(data.Dataset): method __init__ (line 163) | def __init__(self, opt): method __getitem__ (line 199) | def __getitem__(self, index): method __len__ (line 274) | def __len__(self): class Dataset_DefocusDeblur_DualPixel_16bit (line 277) | class Dataset_DefocusDeblur_DualPixel_16bit(data.Dataset): method __init__ (line 278) | def __init__(self, opt): method __getitem__ (line 300) | def __getitem__(self, index): method __len__ (line 363) | def __len__(self): FILE: SRGAN/VmambaIR/data/diffir_dataset.py class DiffIRGANDataset (line 17) | class DiffIRGANDataset(data.Dataset): method __init__ (line 35) | def __init__(self, opt): method __getitem__ (line 57) | def __getitem__(self, index): method __len__ (line 108) | def __len__(self): FILE: SRGAN/VmambaIR/data/diffir_paired_dataset.py class DiffIRGANPairedDataset (line 11) | class DiffIRGANPairedDataset(data.Dataset): method __init__ (line 41) | def __init__(self, opt): method __getitem__ (line 75) | def __getitem__(self, index): method __len__ (line 107) | def __len__(self): FILE: SRGAN/VmambaIR/data/gaussiandenoising_paired_dataset.py class GaussianDenoisingPairedDataset (line 18) | class GaussianDenoisingPairedDataset(data.Dataset): method __init__ (line 46) | def __init__(self, opt): method __getitem__ (line 82) | def __getitem__(self, index): method __len__ (line 158) | def __len__(self): class Dataset_DefocusDeblur_DualPixel_16bit (line 161) | class Dataset_DefocusDeblur_DualPixel_16bit(data.Dataset): method __init__ (line 162) | def __init__(self, opt): method __getitem__ (line 184) | def __getitem__(self, index): method __len__ (line 247) | def __len__(self): FILE: SRGAN/VmambaIR/data/transforms.py function mod_crop (line 5) | def mod_crop(img, scale): function paired_random_crop (line 24) | def paired_random_crop(img_gts, img_lqs, lq_patch_size, scale, gt_path): function paired_random_crop_DP (line 85) | def paired_random_crop_DP(img_lqLs, img_lqRs, img_gts, gt_patch_size, sc... function augment (line 136) | def augment(imgs, hflip=True, rotation=True, flows=None, return_status=F... function img_rotate (line 203) | def img_rotate(img, angle, center=None, scale=1.0): function data_augmentation (line 223) | def data_augmentation(image, mode): function random_augmentation (line 270) | def random_augmentation(*args): FILE: SRGAN/VmambaIR/losses/my_loss.py class KDLoss (line 8) | class KDLoss(nn.Module): method __init__ (line 14) | def __init__(self, loss_weight=1.0, temperature = 0.15): method forward (line 20) | def forward(self, S1_fea, S2_fea): FILE: SRGAN/VmambaIR/models/MambaSISR2_model.py class MambaSISRModel2 (line 16) | class MambaSISRModel2(SRModel): method __init__ (line 24) | def __init__(self, opt): method setup_schedulers (line 28) | def setup_schedulers(self): method feed_data (line 77) | def feed_data(self, data): method nondist_validation (line 81) | def nondist_validation(self, dataloader, current_iter, tb_logger, save... method pad_test (line 87) | def pad_test(self, window_size): method test (line 99) | def test(self): method optimize_parameters (line 195) | def optimize_parameters(self, current_iter): FILE: SRGAN/VmambaIR/models/MambaSISRGAN_model.py class MambaSISRGANModel (line 19) | class MambaSISRGANModel(SRGANModel): method __init__ (line 27) | def __init__(self, opt): method init_training_settings (line 54) | def init_training_settings(self): method feed_data (line 65) | def feed_data(self, data): method nondist_validation (line 70) | def nondist_validation(self, dataloader, current_iter, tb_logger, save... method pad_test (line 76) | def pad_test(self, window_size): method test (line 88) | def test(self): method optimize_parameters (line 110) | def optimize_parameters(self, current_iter): FILE: SRGAN/VmambaIR/models/MambaSISR_model.py class MambaSISRModel (line 16) | class MambaSISRModel(SRModel): method __init__ (line 24) | def __init__(self, opt): method setup_schedulers (line 28) | def setup_schedulers(self): method feed_data (line 77) | def feed_data(self, data): method nondist_validation (line 81) | def nondist_validation(self, dataloader, current_iter, tb_logger, save... method pad_test (line 87) | def pad_test(self, window_size): method test (line 99) | def test(self): method optimize_parameters (line 120) | def optimize_parameters(self, current_iter): FILE: SRGAN/VmambaIR/models/lr_scheduler.py class MultiStepRestartLR (line 7) | class MultiStepRestartLR(_LRScheduler): method __init__ (line 20) | def __init__(self, method get_lr (line 35) | def get_lr(self): class LinearLR (line 49) | class LinearLR(_LRScheduler): method __init__ (line 59) | def __init__(self, method get_lr (line 66) | def get_lr(self): class VibrateLR (line 72) | class VibrateLR(_LRScheduler): method __init__ (line 82) | def __init__(self, method get_lr (line 89) | def get_lr(self): function get_position_from_periods (line 115) | def get_position_from_periods(iteration, cumulative_period): class CosineAnnealingRestartLR (line 136) | class CosineAnnealingRestartLR(_LRScheduler): method __init__ (line 156) | def __init__(self, method get_lr (line 172) | def get_lr(self): class CosineAnnealingRestartCyclicLR (line 186) | class CosineAnnealingRestartCyclicLR(_LRScheduler): method __init__ (line 203) | def __init__(self, method get_lr (line 219) | def get_lr(self): FILE: SRGAN/VmambaIR/train_pipeline.py function init_tb_loggers (line 18) | def init_tb_loggers(opt): function create_train_val_dataloader (line 30) | def create_train_val_dataloader(opt, logger): function load_resume_state (line 69) | def load_resume_state(opt): function train_pipeline (line 92) | def train_pipeline(root_path): FILE: SRGAN/VmambaIR/utils.py class RealESRGANer (line 14) | class RealESRGANer(): method __init__ (line 29) | def __init__(self, method pre_process (line 68) | def pre_process(self, img): method process (line 93) | def process(self): method tile_process (line 97) | def tile_process(self): method post_process (line 162) | def post_process(self): method enhance (line 174) | def enhance(self, img, outscale=None, alpha_upsampler='realesrgan'): class PrefetchReader (line 246) | class PrefetchReader(threading.Thread): method __init__ (line 254) | def __init__(self, img_list, num_prefetch_queue): method run (line 259) | def run(self): method __next__ (line 266) | def __next__(self): method __iter__ (line 272) | def __iter__(self): class IOConsumer (line 276) | class IOConsumer(threading.Thread): method __init__ (line 278) | def __init__(self, opt, que, qid): method run (line 284) | def run(self): FILE: SRGAN/VmambaIR/utils/bundle_submissions.py function bundle_submissions_raw (line 13) | def bundle_submissions_raw(submission_folder,session): function bundle_submissions_srgb (line 46) | def bundle_submissions_srgb(submission_folder,session): function bundle_submissions_srgb_v1 (line 79) | def bundle_submissions_srgb_v1(submission_folder,session): FILE: SRGAN/VmambaIR/utils/create_lmdb.py function prepare_keys (line 7) | def prepare_keys(folder_path, suffix='png'): function create_lmdb_for_reds (line 24) | def create_lmdb_for_reds(): function create_lmdb_for_gopro (line 46) | def create_lmdb_for_gopro(): function create_lmdb_for_rain13k (line 71) | def create_lmdb_for_rain13k(): function create_lmdb_for_SIDD (line 84) | def create_lmdb_for_SIDD(): FILE: SRGAN/VmambaIR/utils/dist_util.py function init_dist (line 10) | def init_dist(launcher, backend='nccl', **kwargs): function _init_dist_pytorch (line 21) | def _init_dist_pytorch(backend, **kwargs): function _init_dist_slurm (line 28) | def _init_dist_slurm(backend, port=None): function get_dist_info (line 61) | def get_dist_info(): function master_only (line 75) | def master_only(func): FILE: SRGAN/VmambaIR/utils/download_util.py function download_file_from_google_drive (line 8) | def download_file_from_google_drive(file_id, save_path): function get_confirm_token (line 41) | def get_confirm_token(response): function save_response_content (line 48) | def save_response_content(response, FILE: SRGAN/VmambaIR/utils/face_util.py class FaceRestorationHelper (line 16) | class FaceRestorationHelper(object): method __init__ (line 19) | def __init__(self, upscale_factor, face_size=512): method init_dlib (line 41) | def init_dlib(self, detection_path, landmark5_path, landmark68_path): method free_dlib_gpu_memory (line 47) | def free_dlib_gpu_memory(self): method read_input_image (line 52) | def read_input_image(self, img_path): method detect_faces (line 56) | def detect_faces(self, method get_face_landmarks_5 (line 89) | def get_face_landmarks_5(self): method get_face_landmarks_68 (line 96) | def get_face_landmarks_68(self): method warp_crop_faces (line 132) | def warp_crop_faces(self, method add_restored_face (line 170) | def add_restored_face(self, face): method paste_faces_to_input_image (line 173) | def paste_faces_to_input_image(self, save_path): method clean_all (line 211) | def clean_all(self): FILE: SRGAN/VmambaIR/utils/file_client.py class BaseStorageBackend (line 5) | class BaseStorageBackend(metaclass=ABCMeta): method get (line 14) | def get(self, filepath): method get_text (line 18) | def get_text(self, filepath): class MemcachedBackend (line 22) | class MemcachedBackend(BaseStorageBackend): method __init__ (line 32) | def __init__(self, server_list_cfg, client_cfg, sys_path=None): method get (line 49) | def get(self, filepath): method get_text (line 56) | def get_text(self, filepath): class HardDiskBackend (line 60) | class HardDiskBackend(BaseStorageBackend): method get (line 63) | def get(self, filepath): method get_text (line 69) | def get_text(self, filepath): class LmdbBackend (line 76) | class LmdbBackend(BaseStorageBackend): method __init__ (line 96) | def __init__(self, method get (line 131) | def get(self, filepath, client_key): method get_text (line 146) | def get_text(self, filepath): class FileClient (line 150) | class FileClient(object): method __init__ (line 169) | def __init__(self, backend='disk', **kwargs): method get (line 177) | def get(self, filepath, client_key='default'): method get_text (line 185) | def get_text(self, filepath): FILE: SRGAN/VmambaIR/utils/flow_util.py function flowread (line 7) | def flowread(flow_path, quantize=False, concat_axis=0, *args, **kwargs): function flowwrite (line 47) | def flowwrite(flow, filename, quantize=False, concat_axis=0, *args, **kw... function quantize_flow (line 78) | def quantize_flow(flow, max_val=0.02, norm=True): function dequantize_flow (line 106) | def dequantize_flow(dx, dy, max_val=0.02, denorm=True): function quantize (line 130) | def quantize(arr, min_val, max_val, levels, dtype=np.int64): function dequantize (line 157) | def dequantize(arr, min_val, max_val, levels, dtype=np.float64): FILE: SRGAN/VmambaIR/utils/img_util.py function img2tensor (line 9) | def img2tensor(imgs, bgr2rgb=True, float32=True): function tensor2img (line 36) | def tensor2img(tensor, rgb2bgr=True, out_type=np.uint8, min_max=(0, 1)): function imfrombytes (line 101) | def imfrombytes(content, flag='color', float32=False): function imfrombytesDP (line 127) | def imfrombytesDP(content, flag='color', float32=False): function padding (line 148) | def padding(img_lq, img_gt, gt_size): function padding_DP (line 166) | def padding_DP(img_lqL, img_lqR, img_gt, gt_size): function imwrite (line 181) | def imwrite(img, file_path, params=None, auto_mkdir=True): function crop_border (line 200) | def crop_border(imgs, crop_border): FILE: SRGAN/VmambaIR/utils/lmdb_util.py function make_lmdb_from_imgs (line 9) | def make_lmdb_from_imgs(data_path, function read_img_worker (line 139) | def read_img_worker(path, key, compress_level): class LmdbMaker (line 164) | class LmdbMaker(): method __init__ (line 175) | def __init__(self, method put (line 194) | def put(self, img_byte, key, img_shape): method close (line 205) | def close(self): FILE: SRGAN/VmambaIR/utils/logger.py class MessageLogger (line 10) | class MessageLogger(): method __init__ (line 23) | def __init__(self, opt, start_iter=1, tb_logger=None): method __call__ (line 34) | def __call__(self, log_vars): function init_tb_logger (line 81) | def init_tb_logger(log_dir): function init_wandb_logger (line 88) | def init_wandb_logger(opt): function get_root_logger (line 108) | def get_root_logger(logger_name='basicsr', log_level=logging.INFO, log_f... function get_env_info (line 150) | def get_env_info(): FILE: SRGAN/VmambaIR/utils/matlab_functions.py function cubic (line 6) | def cubic(x): function calculate_weights_indices (line 17) | def calculate_weights_indices(in_length, out_length, scale, kernel, function imresize (line 88) | def imresize(img, scale, antialiasing=True): function rgb2ycbcr (line 173) | def rgb2ycbcr(img, y_only=False): function bgr2ycbcr (line 207) | def bgr2ycbcr(img, y_only=False): function ycbcr2rgb (line 241) | def ycbcr2rgb(img): function ycbcr2bgr (line 273) | def ycbcr2bgr(img): function _convert_input_type_range (line 305) | def _convert_input_type_range(img): function _convert_output_type_range (line 333) | def _convert_output_type_range(img, dst_type): FILE: SRGAN/VmambaIR/utils/misc.py function set_random_seed (line 12) | def set_random_seed(seed): function get_time_str (line 21) | def get_time_str(): function mkdir_and_rename (line 25) | def mkdir_and_rename(path): function make_exp_dirs (line 39) | def make_exp_dirs(opt): function scandir (line 53) | def scandir(dir_path, suffix=None, recursive=False, full_path=False): function scandir_SIDD (line 95) | def scandir_SIDD(dir_path, keywords=None, recursive=False, full_path=Fal... function check_resume (line 137) | def check_resume(opt, resume_iter): function sizeof_fmt (line 166) | def sizeof_fmt(size, suffix='B'): FILE: SRGAN/VmambaIR/utils/options.py function ordered_yaml (line 6) | def ordered_yaml(): function parse (line 31) | def parse(opt_path, is_train=True): function dict2str (line 92) | def dict2str(opt, indent_level=1): FILE: SRGAN/ldm/classifier.py function disabled_train (line 22) | def disabled_train(self, mode=True): class NoisyLatentImageClassifier (line 28) | class NoisyLatentImageClassifier(pl.LightningModule): method __init__ (line 30) | def __init__(self, method init_from_ckpt (line 70) | def init_from_ckpt(self, path, ignore_keys=list(), only_model=False): method load_diffusion (line 88) | def load_diffusion(self): method load_classifier (line 95) | def load_classifier(self, ckpt_path, pool): method get_x_noisy (line 110) | def get_x_noisy(self, x, t, noise=None): method forward (line 120) | def forward(self, x_noisy, t, *args, **kwargs): method get_input (line 124) | def get_input(self, batch, k): method get_conditioning (line 133) | def get_conditioning(self, batch, k=None): method compute_top_k (line 150) | def compute_top_k(self, logits, labels, k, reduction="mean"): method on_train_epoch_start (line 157) | def on_train_epoch_start(self): method write_logs (line 162) | def write_logs(self, loss, logits, targets): method shared_step (line 179) | def shared_step(self, batch, t=None): method training_step (line 198) | def training_step(self, batch, batch_idx): method reset_noise_accs (line 202) | def reset_noise_accs(self): method on_validation_start (line 206) | def on_validation_start(self): method validation_step (line 210) | def validation_step(self, batch, batch_idx): method configure_optimizers (line 220) | def configure_optimizers(self): method log_images (line 238) | def log_images(self, batch, N=8, *args, **kwargs): FILE: SRGAN/ldm/lr_scheduler.py class LambdaWarmUpCosineScheduler (line 4) | class LambdaWarmUpCosineScheduler: method __init__ (line 8) | def __init__(self, warm_up_steps, lr_min, lr_max, lr_start, max_decay_... method schedule (line 17) | def schedule(self, n, **kwargs): method __call__ (line 32) | def __call__(self, n, **kwargs): class LambdaWarmUpCosineScheduler2 (line 36) | class LambdaWarmUpCosineScheduler2: method __init__ (line 41) | def __init__(self, warm_up_steps, f_min, f_max, f_start, cycle_lengths... method find_in_interval (line 52) | def find_in_interval(self, n): method schedule (line 59) | def schedule(self, n, **kwargs): method __call__ (line 77) | def __call__(self, n, **kwargs): class LambdaLinearScheduler (line 81) | class LambdaLinearScheduler(LambdaWarmUpCosineScheduler2): method schedule (line 83) | def schedule(self, n, **kwargs): FILE: SRGAN/ldm/util.py function log_txt_as_img (line 17) | def log_txt_as_img(wh, xc, size=10): function ismap (line 41) | def ismap(x): function isimage (line 47) | def isimage(x): function exists (line 53) | def exists(x): function default (line 57) | def default(val, d): function mean_flat (line 63) | def mean_flat(tensor): function count_params (line 71) | def count_params(model, verbose=False): function instantiate_from_config (line 78) | def instantiate_from_config(config): function get_obj_from_str (line 88) | def get_obj_from_str(string, reload=False): function _do_parallel_data_prefetch (line 96) | def _do_parallel_data_prefetch(func, Q, data, idx, idx_to_fn=False): function parallel_data_prefetch (line 108) | def parallel_data_prefetch( FILE: SRGAN/ldm/util2.py function make_beta_schedule (line 21) | def make_beta_schedule(schedule, n_timestep, linear_start=1e-4, linear_e... function make_ddim_timesteps (line 46) | def make_ddim_timesteps(ddim_discr_method, num_ddim_timesteps, num_ddpm_... function make_ddim_sampling_parameters (line 63) | def make_ddim_sampling_parameters(alphacums, ddim_timesteps, eta, verbos... function betas_for_alpha_bar (line 77) | def betas_for_alpha_bar(num_diffusion_timesteps, alpha_bar, max_beta=0.9... function extract_into_tensor (line 96) | def extract_into_tensor(a, t, x_shape): function checkpoint (line 102) | def checkpoint(func, inputs, params, flag): class CheckpointFunction (line 119) | class CheckpointFunction(torch.autograd.Function): method forward (line 121) | def forward(ctx, run_function, length, *args): method backward (line 131) | def backward(ctx, *output_grads): function timestep_embedding (line 151) | def timestep_embedding(timesteps, dim, max_period=10000, repeat_only=Fal... function zero_module (line 174) | def zero_module(module): function scale_module (line 183) | def scale_module(module, scale): function mean_flat (line 192) | def mean_flat(tensor): function normalization (line 199) | def normalization(channels): class SiLU (line 209) | class SiLU(nn.Module): method forward (line 210) | def forward(self, x): class GroupNorm32 (line 214) | class GroupNorm32(nn.GroupNorm): method forward (line 215) | def forward(self, x): function conv_nd (line 218) | def conv_nd(dims, *args, **kwargs): function linear (line 231) | def linear(*args, **kwargs): function avg_pool_nd (line 238) | def avg_pool_nd(dims, *args, **kwargs): class HybridConditioner (line 251) | class HybridConditioner(nn.Module): method __init__ (line 253) | def __init__(self, c_concat_config, c_crossattn_config): method forward (line 258) | def forward(self, c_concat, c_crossattn): function noise_like (line 264) | def noise_like(shape, device, repeat=False): FILE: SRGAN/scripts/extract_subimages.py function main (line 12) | def main(args): function extract_subimages (line 43) | def extract_subimages(opt): function worker (line 74) | def worker(path, opt): FILE: SRGAN/scripts/extract_subimages_DF2K.py function main (line 12) | def main(args): function extract_subimages (line 43) | def extract_subimages(opt): function worker (line 74) | def worker(path, opt): FILE: SRGAN/scripts/generate_meta_info.py function main (line 7) | def main(args): FILE: SRGAN/scripts/generate_meta_info_DF2K.py function main (line 7) | def main(args): FILE: SRGAN/scripts/generate_meta_info_OST.py function main (line 7) | def main(args): FILE: SRGAN/scripts/generate_meta_info_pairdata.py function main (line 6) | def main(args): FILE: SRGAN/scripts/generate_multiscale_DF2K.py function main (line 7) | def main(args): FILE: SRGAN/scripts/pytorch2onnx.py function main (line 7) | def main(args): FILE: SRGAN/setup.py function readme (line 12) | def readme(): function get_git_hash (line 18) | def get_git_hash(): function get_hash (line 43) | def get_hash(): function write_version_py (line 52) | def write_version_py(): function get_version (line 69) | def get_version(): function get_requirements (line 75) | def get_requirements(filename='requirements.txt'):