SYMBOL INDEX (1285 symbols across 125 files) FILE: code/real/bsrt/data_processing/camera_pipeline.py function random_ccm (line 13) | def random_ccm(): function random_gains (line 47) | def random_gains(): function apply_smoothstep (line 58) | def apply_smoothstep(image): function invert_smoothstep (line 64) | def invert_smoothstep(image): function gamma_expansion (line 70) | def gamma_expansion(image): function gamma_compression (line 76) | def gamma_compression(image): function apply_ccm (line 82) | def apply_ccm(image, ccm): function apply_gains (line 95) | def apply_gains(image, rgb_gain, red_gain, blue_gain): function safe_invert_gains (line 109) | def safe_invert_gains(image, rgb_gain, red_gain, blue_gain): function mosaic (line 125) | def mosaic(image, mode='rggb'): function demosaic (line 151) | def demosaic(image): function random_noise_levels (line 188) | def random_noise_levels(): function add_noise (line 201) | def add_noise(image, shot_noise=0.01, read_noise=0.0005): function process_linear_image_rgb (line 208) | def process_linear_image_rgb(image, meta_info, return_np=False): function process_linear_image_raw (line 225) | def process_linear_image_raw(image, meta_info): FILE: code/real/bsrt/data_processing/synthetic_burst_generation.py function random_crop (line 10) | def random_crop(frames, crop_sz): function rgb2rawburst (line 49) | def rgb2rawburst(image, burst_size, downsample_factor=1, burst_transform... function get_tmat (line 123) | def get_tmat(image_shape, translation, theta, shear_values, scale_factors): function single2lrburst (line 149) | def single2lrburst(image, burst_size, downsample_factor=1, transformatio... FILE: code/real/bsrt/datasets/burstsr_dataset.py class SamsungRAWImage (line 10) | class SamsungRAWImage: method load (line 12) | def load(path): method __init__ (line 24) | def __init__(self, im_raw, black_level, cam_wb, daylight_wb, color_mat... method get_all_meta_data (line 38) | def get_all_meta_data(self): method get_exposure_time (line 42) | def get_exposure_time(self): method get_noise_profile (line 45) | def get_noise_profile(self): method get_f_number (line 51) | def get_f_number(self): method get_iso (line 54) | def get_iso(self): method get_image_data (line 57) | def get_image_data(self, substract_black_level=False, white_balance=Fa... method shape (line 72) | def shape(self): method crop_image (line 76) | def crop_image(self, r1, r2, c1, c2): method get_crop (line 79) | def get_crop(self, r1, r2, c1, c2): method postprocess (line 90) | def postprocess(self, return_np=True, norm_factor=None): class CanonImage (line 113) | class CanonImage: method load (line 115) | def load(path, split='train'): method __init__ (line 125) | def __init__(self, im_raw, black_level, cam_wb, daylight_wb, rgb_xyz_m... method shape (line 150) | def shape(self): method get_all_meta_data (line 154) | def get_all_meta_data(self): method get_exposure_time (line 159) | def get_exposure_time(self): method get_f_number (line 162) | def get_f_number(self): method get_iso (line 165) | def get_iso(self): method get_image_data (line 168) | def get_image_data(self, substract_black_level=False, white_balance=Fa... method set_image_data (line 182) | def set_image_data(self, im_data): method crop_image (line 185) | def crop_image(self, r1, r2, c1, c2): method get_crop (line 188) | def get_crop(self, r1, r2, c1, c2): method set_crop_info (line 193) | def set_crop_info(self, crop_info): method resize (line 196) | def resize(self, size=None, scale_factor=None): method postprocess (line 201) | def postprocess(self, return_np=True): function load_txt (line 216) | def load_txt(path): class BurstSRDataset (line 223) | class BurstSRDataset(torch.utils.data.Dataset): method __init__ (line 225) | def __init__(self, root, burst_size=8, crop_sz=80, center_crop=False, ... method _get_burst_list (line 255) | def _get_burst_list(self): method get_burst_info (line 260) | def get_burst_info(self, burst_id): method _get_raw_image (line 264) | def _get_raw_image(self, burst_id, im_id): method _get_gt_image (line 268) | def _get_gt_image(self, burst_id): method get_burst (line 272) | def get_burst(self, burst_id, im_ids, info=None): method _sample_images (line 281) | def _sample_images(self): method __len__ (line 288) | def __len__(self): method __getitem__ (line 291) | def __getitem__(self, index): function pack_raw_image (line 400) | def pack_raw_image(im_raw): function flatten_raw_image (line 415) | def flatten_raw_image(im_raw_4ch): function pack_raw_image_batch (line 430) | def pack_raw_image_batch(im_raw): function flatten_raw_image_batch (line 439) | def flatten_raw_image_batch(im_raw_4ch): FILE: code/real/bsrt/datasets/burstsr_test_dataset.py class BurstSRDataset (line 8) | class BurstSRDataset(torch.utils.data.Dataset): method __init__ (line 10) | def __init__(self, root, burst_size=8, crop_sz=80, center_crop=False, ... method _get_burst_list (line 40) | def _get_burst_list(self): method get_burst_info (line 45) | def get_burst_info(self, burst_id): method _get_raw_image (line 49) | def _get_raw_image(self, burst_id, im_id): method get_burst (line 53) | def get_burst(self, burst_id, im_ids, info=None): method _sample_images (line 61) | def _sample_images(self): method __len__ (line 68) | def __len__(self): method __getitem__ (line 71) | def __getitem__(self, index): FILE: code/real/bsrt/datasets/data_sampler.py class DistIterSampler (line 13) | class DistIterSampler(Sampler): method __init__ (line 31) | def __init__(self, dataset, num_replicas=None, rank=None, ratio=100): method __iter__ (line 47) | def __iter__(self): method __len__ (line 64) | def __len__(self): method set_epoch (line 67) | def set_epoch(self, epoch): FILE: code/real/bsrt/datasets/realworld_burst_test_set.py class RealWorldBurstTest (line 7) | class RealWorldBurstTest(torch.utils.data.Dataset): method __init__ (line 10) | def __init__(self, root): method __len__ (line 15) | def __len__(self): method _read_burst_image (line 18) | def _read_burst_image(self, index, image_id): method __getitem__ (line 23) | def __getitem__(self, index): FILE: code/real/bsrt/datasets/synthetic_burst_test_set.py class SyntheticBurstTest (line 7) | class SyntheticBurstTest(torch.utils.data.Dataset): method __init__ (line 11) | def __init__(self, root): method __len__ (line 16) | def __len__(self): method _read_burst_image (line 19) | def _read_burst_image(self, index, image_id): method __getitem__ (line 24) | def __getitem__(self, index): FILE: code/real/bsrt/datasets/synthetic_burst_train_set.py class SyntheticBurst (line 8) | class SyntheticBurst(torch.utils.data.Dataset): method __init__ (line 18) | def __init__(self, base_dataset, burst_size=8, crop_sz=384, transform=... method __len__ (line 37) | def __len__(self): method __getitem__ (line 40) | def __getitem__(self, index): FILE: code/real/bsrt/datasets/synthetic_burst_val_set.py class SyntheticBurstVal (line 8) | class SyntheticBurstVal(torch.utils.data.Dataset): method __init__ (line 15) | def __init__(self, root=None, initialize=True): method initialize (line 25) | def initialize(self): method __len__ (line 28) | def __len__(self): method _read_burst_image (line 31) | def _read_burst_image(self, index, image_id): method _read_gt_image (line 37) | def _read_gt_image(self, index): method _read_meta_info (line 42) | def _read_meta_info(self, index): method __getitem__ (line 48) | def __getitem__(self, index): FILE: code/real/bsrt/datasets/zurich_raw2rgb_dataset.py class ZurichRAW2RGB (line 7) | class ZurichRAW2RGB(torch.utils.data.Dataset): method __init__ (line 12) | def __init__(self, root, split='train'): method _get_image_list (line 22) | def _get_image_list(self, split): method _get_image (line 33) | def _get_image(self, im_id): method get_image (line 38) | def get_image(self, im_id): method __len__ (line 43) | def __len__(self): method __getitem__ (line 46) | def __getitem__(self, index): FILE: code/real/bsrt/loss/Charbonnier.py class CharbonnierLoss (line 5) | class CharbonnierLoss(nn.Module): method __init__ (line 8) | def __init__(self, epsilon=1e-3, reduce=True): method forward (line 13) | def forward(self, X, Y): FILE: code/real/bsrt/loss/__init__.py class Loss (line 14) | class Loss(nn.modules.loss._Loss): method __init__ (line 15) | def __init__(self, args, ckp): method forward (line 80) | def forward(self, sr, hr): method step (line 97) | def step(self): method start_log (line 102) | def start_log(self): method end_log (line 105) | def end_log(self, n_batches): method display_loss (line 108) | def display_loss(self, batch): method plot_loss (line 116) | def plot_loss(self, apath, epoch): method get_loss_module (line 130) | def get_loss_module(self): method save (line 136) | def save(self, apath): method load (line 140) | def load(self, apath, cpu=False): FILE: code/real/bsrt/loss/adversarial.py class Adversarial (line 12) | class Adversarial(nn.Module): method __init__ (line 13) | def __init__(self, args, gan_type): method forward (line 36) | def forward(self, fake, real): method state_dict (line 96) | def state_dict(self, *args, **kwargs): method bce (line 102) | def bce(self, real, fake): FILE: code/real/bsrt/loss/discriminator.py class Discriminator (line 5) | class Discriminator(nn.Module): method __init__ (line 9) | def __init__(self, args, gan_type='GAN'): method forward (line 65) | def forward(self, x): FILE: code/real/bsrt/loss/filter.py class Filter (line 5) | class Filter(nn.Module): method __init__ (line 6) | def __init__(self, args): method forward (line 16) | def forward(self, x, y): FILE: code/real/bsrt/loss/hist_entropy.py class HistEntropy (line 5) | class HistEntropy(nn.Module): method __init__ (line 6) | def __init__(self, args): method forward (line 10) | def forward(self, x): FILE: code/real/bsrt/loss/mssim.py function gaussian (line 7) | def gaussian(window_size, sigma): function create_window (line 12) | def create_window(window_size, channel=1): function ssim (line 19) | def ssim(img1, img2, window_size=11, window=None, size_average=True, ful... function msssim (line 71) | def msssim(img1, img2, window_size=11, size_average=True, val_range=None... class SSIM (line 109) | class SSIM(torch.nn.Module): method __init__ (line 110) | def __init__(self, window_size=11, size_average=True, val_range=None): method forward (line 120) | def forward(self, img1, img2): class MSSSIM (line 132) | class MSSSIM(torch.nn.Module): method __init__ (line 133) | def __init__(self, window_size=11, size_average=True, channel=3): method forward (line 139) | def forward(self, img1, img2): FILE: code/real/bsrt/loss/vgg.py class VGG (line 8) | class VGG(nn.Module): method __init__ (line 9) | def __init__(self, conv_index, rgb_range=1): method forward (line 24) | def forward(self, sr, hr): FILE: code/real/bsrt/main.py function init_seeds (line 19) | def init_seeds(seed=0, cuda_deterministic=True): function main (line 33) | def main(): function main_worker (line 37) | def main_worker(local_rank, nprocs, args): FILE: code/real/bsrt/model/DCNv2/dcn_v2.py class _DCNv2 (line 17) | class _DCNv2(Function): method forward (line 21) | def forward( method backward (line 52) | def backward(ctx, grad_output): method symbolic (line 75) | def symbolic( class DCNv2 (line 102) | class DCNv2(nn.Module): method __init__ (line 103) | def __init__( method reset_parameters (line 126) | def reset_parameters(self): method forward (line 134) | def forward(self, input, offset, mask): class DCN (line 153) | class DCN(DCNv2): method __init__ (line 154) | def __init__( method init_offset (line 179) | def init_offset(self): method forward (line 183) | def forward(self, input): class DCN_sep (line 201) | class DCN_sep(DCNv2): method __init__ (line 204) | def __init__(self, method init_offset (line 225) | def init_offset(self): method forward (line 229) | def forward(self, input, fea): class FlowGuidedDCN (line 249) | class FlowGuidedDCN(DCNv2): method __init__ (line 252) | def __init__(self, method init_offset (line 269) | def init_offset(self): method forward (line 273) | def forward(self, input, fea, flows): class InsideFlowGuidedDCN (line 296) | class InsideFlowGuidedDCN(DCNv2): method __init__ (line 299) | def __init__(self, method reset_parameters (line 322) | def reset_parameters(self): method init_offset (line 331) | def init_offset(self): method forward (line 335) | def forward(self, input, warped, ref, flows): class _DCNv2Pooling (line 359) | class _DCNv2Pooling(Function): method forward (line 361) | def forward( method backward (line 402) | def backward(ctx, grad_output): class DCNv2Pooling (line 426) | class DCNv2Pooling(nn.Module): method __init__ (line 427) | def __init__( method forward (line 448) | def forward(self, input, rois, offset): class DCNPooling (line 467) | class DCNPooling(DCNv2Pooling): method __init__ (line 468) | def __init__( method forward (line 506) | def forward(self, input, rois): FILE: code/real/bsrt/model/DCNv2/setup.py function get_extensions (line 13) | def get_extensions(): FILE: code/real/bsrt/model/DCNv2/src/cpu/dcn_v2_cpu.cpp function dcn_v2_cpu_forward (line 17) | at::Tensor function dcn_v2_cpu_backward (line 109) | std::vector dcn_v2_cpu_backward(const at::Tensor &input, FILE: code/real/bsrt/model/DCNv2/src/cpu/dcn_v2_im2col_cpu.cpp function dmcn_im2col_bilinear_cpu (line 27) | float dmcn_im2col_bilinear_cpu(const float *bottom_data, const int data_... function dmcn_get_gradient_weight_cpu (line 58) | float dmcn_get_gradient_weight_cpu(float argmax_h, float argmax_w, function dmcn_get_coordinate_weight_cpu (line 84) | float dmcn_get_coordinate_weight_cpu(float argmax_h, float argmax_w, function modulated_deformable_im2col_cpu_kernel (line 127) | void modulated_deformable_im2col_cpu_kernel(const int n, const float *da... function modulated_deformable_col2im_cpu_kernel (line 198) | void modulated_deformable_col2im_cpu_kernel(const int n, const float *da... function modulated_deformable_col2im_coord_cpu_kernel (line 259) | void modulated_deformable_col2im_coord_cpu_kernel(const int n, const flo... function modulated_deformable_im2col_cpu (line 331) | void modulated_deformable_im2col_cpu(const float* data_im, const float* ... function modulated_deformable_col2im_cpu (line 353) | void modulated_deformable_col2im_cpu(const float* data_col, const float*... function modulated_deformable_col2im_coord_cpu (line 375) | void modulated_deformable_col2im_coord_cpu(const float* data_col, const ... FILE: code/real/bsrt/model/DCNv2/src/cpu/dcn_v2_psroi_pooling_cpu.cpp function T (line 34) | T bilinear_interp_cpu( function DeformablePSROIPoolForwardKernelCpu (line 59) | void DeformablePSROIPoolForwardKernelCpu( function DeformablePSROIPoolBackwardAccKernelCpu (line 149) | void DeformablePSROIPoolBackwardAccKernelCpu( function dcn_v2_psroi_pooling_cpu_forward (line 278) | std::tuple function dcn_v2_psroi_pooling_cpu_backward (line 350) | std::tuple FILE: code/real/bsrt/model/DCNv2/src/vision.cpp function PYBIND11_MODULE (line 4) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) { FILE: code/real/bsrt/model/DCNv2/test.py function conv_identify (line 20) | def conv_identify(weight, bias): function check_zero_offset (line 32) | def check_zero_offset(): function check_gradient_dconv (line 69) | def check_gradient_dconv(): function check_pooling_zero_offset (line 100) | def check_pooling_zero_offset(): function check_gradient_dpooling (line 134) | def check_gradient_dpooling(): function example_dconv (line 169) | def example_dconv(): function example_dpooling (line 183) | def example_dpooling(): function example_mdpooling (line 226) | def example_mdpooling(): FILE: code/real/bsrt/model/__init__.py class Model (line 10) | class Model(nn.Module): method __init__ (line 11) | def __init__(self, args, ckp): method forward (line 52) | def forward(self, x, idx_scale): method save (line 72) | def save(self, apath, epoch, is_best=False): method load (line 89) | def load(self, apath, pre_train='', resume=-1, cpu=False): method forward_chop (line 219) | def forward_chop(self, *args, shave=10, min_size=160000): method forward_x8 (line 278) | def forward_x8(self, *args, forward_function=None): FILE: code/real/bsrt/model/arch_util.py function initialize_weights (line 9) | def initialize_weights(net_l, scale=1): function make_layer (line 29) | def make_layer(block, n_layers): function conv_layer (line 38) | def conv_layer(in_channels, out_channels, kernel_size, stride=1, padding... class ESA (line 42) | class ESA(nn.Module): method __init__ (line 43) | def __init__(self, n_feats, conv=conv_layer): method forward (line 56) | def forward(self, x): class DWConv (line 71) | class DWConv(nn.Module): method __init__ (line 72) | def __init__(self, dim=768): method forward (line 76) | def forward(self, x): class SELayer (line 82) | class SELayer(nn.Module): method __init__ (line 86) | def __init__(self, channel, reduction=16): method forward (line 96) | def forward(self, x): class ResidualBlock_noBN (line 102) | class ResidualBlock_noBN(nn.Module): method __init__ (line 108) | def __init__(self, nf=64): method forward (line 116) | def forward(self, x): class ResidualBlock_SE (line 123) | class ResidualBlock_SE(nn.Module): method __init__ (line 129) | def __init__(self, nf=64, reduction=16): method forward (line 138) | def forward(self, x): class _PositionAttentionModule (line 148) | class _PositionAttentionModule(nn.Module): method __init__ (line 151) | def __init__(self, in_channels, **kwargs): method forward (line 159) | def forward(self, x): class SALayer (line 171) | class SALayer(nn.Module): method __init__ (line 172) | def __init__(self, wn=None): method forward (line 178) | def forward(self, x): class CALayerV2 (line 186) | class CALayerV2(nn.Module): method __init__ (line 187) | def __init__(self, n_feat, reduction=16, wn=None): method forward (line 200) | def forward(self, x): class DALayer (line 207) | class DALayer(nn.Module): method __init__ (line 208) | def __init__(self, channel, reduction, wn): method forward (line 215) | def forward(self, x): class CALayer (line 223) | class CALayer(nn.Module): method __init__ (line 224) | def __init__(self, channel, reduction, wn): method forward (line 236) | def forward(self, x): class RCAB (line 243) | class RCAB(nn.Module): method __init__ (line 244) | def __init__( method forward (line 266) | def forward(self, x): class ResidualGroup (line 273) | class ResidualGroup(nn.Module): method __init__ (line 274) | def __init__(self, n_feat, n_resblocks, da=False): method forward (line 292) | def forward(self, x): function make_layer_idx (line 302) | def make_layer_idx(block, n_layers): class LRSCRCAB (line 309) | class LRSCRCAB(nn.Module): method __init__ (line 310) | def __init__( method forward (line 332) | def forward(self, x): class LRSCPYRCAB (line 339) | class LRSCPYRCAB(nn.Module): method __init__ (line 340) | def __init__( method forward (line 366) | def forward(self, x): class LRSCResidualGroup (line 372) | class LRSCResidualGroup(nn.Module): method __init__ (line 373) | def __init__(self, n_feat, n_resblocks, da=False, idx=0): method forward (line 392) | def forward(self, x): class LRSCPSResidualGroup (line 400) | class LRSCPSResidualGroup(nn.Module): method __init__ (line 401) | def __init__(self, n_feat, n_resblocks, da=False, idx=0): method forward (line 421) | def forward(self, x): class LRSCPyResidualGroup (line 430) | class LRSCPyResidualGroup(nn.Module): method __init__ (line 431) | def __init__(self, n_feat, n_resblocks, da=False, idx=0): method forward (line 451) | def forward(self, x): class LRSCWideActResBlock (line 458) | class LRSCWideActResBlock(nn.Module): method __init__ (line 459) | def __init__(self, nf=64, idx=0): method forward (line 482) | def forward(self, x): class LRSCPyWideActResBlock (line 488) | class LRSCPyWideActResBlock(nn.Module): method __init__ (line 489) | def __init__(self, nf=64, idx=0): method forward (line 515) | def forward(self, x): class LRSCPyWideActResGroup (line 523) | class LRSCPyWideActResGroup(nn.Module): method __init__ (line 524) | def __init__(self, nf, n_resblocks, idx=0): method forward (line 539) | def forward(self, x): class LRSCWideActResGroup (line 548) | class LRSCWideActResGroup(nn.Module): method __init__ (line 549) | def __init__(self, nf, n_resblocks, idx=0): method forward (line 564) | def forward(self, x): class PYRCAB (line 577) | class PYRCAB(nn.Module): method __init__ (line 578) | def __init__( method forward (line 603) | def forward(self, x): class PyResidualGroup (line 609) | class PyResidualGroup(nn.Module): method __init__ (line 610) | def __init__(self, n_feat, n_resblocks, da=False): method forward (line 632) | def forward(self, x): class WideActResBlock (line 637) | class WideActResBlock(nn.Module): method __init__ (line 638) | def __init__(self, nf=64): method forward (line 658) | def forward(self, x): class PSWideActResBlock (line 664) | class PSWideActResBlock(nn.Module): method __init__ (line 665) | def __init__(self, nf=64): method forward (line 685) | def forward(self, x): class PyWideActResBlock (line 691) | class PyWideActResBlock(nn.Module): method __init__ (line 692) | def __init__(self, nf=64): method forward (line 717) | def forward(self, x): function flow_warp (line 723) | def flow_warp(x, flow, interp_mode='bilinear', padding_mode='zeros', ali... FILE: code/real/bsrt/model/bsrt.py function make_model (line 27) | def make_model(args, parent=False): class BasicModule (line 66) | class BasicModule(nn.Module): method __init__ (line 70) | def __init__(self): method forward (line 80) | def forward(self, tensor_input): class SpyNet (line 84) | class SpyNet(nn.Module): method __init__ (line 92) | def __init__(self, load_path=None, return_levels=[5]): method preprocess (line 110) | def preprocess(self, tensor_input): method process (line 114) | def process(self, ref, supp, w, h, w_floor, h_floor): method forward (line 160) | def forward(self, ref, supp): class FlowGuidedPCDAlign (line 176) | class FlowGuidedPCDAlign(nn.Module): method __init__ (line 181) | def __init__(self, nf=64, groups=8): method forward (line 209) | def forward(self, nbr_fea_l, nbr_fea_warped_l, ref_fea_l, flows_l): class CrossNonLocal_Fusion (line 246) | class CrossNonLocal_Fusion(nn.Module): method __init__ (line 249) | def __init__(self, nf=64, out_feat=96, nframes=5, center=2): method forward (line 265) | def forward(self, aligned_fea): class BSRT (line 287) | class BSRT(nn.Module): method __init__ (line 288) | def __init__(self, args, nframes=8, img_size=64, patch_size=1, in_chan... method _init_weights (line 452) | def _init_weights(self, m): method no_weight_decay (line 462) | def no_weight_decay(self): method no_weight_decay_keywords (line 466) | def no_weight_decay_keywords(self): method _upsample_add (line 469) | def _upsample_add(self, x, y): method check_image_size (line 472) | def check_image_size(self, x): method pre_forward_features (line 479) | def pre_forward_features(self, x): method forward_features (line 498) | def forward_features(self, x): method forward (line 516) | def forward(self, x, print_time=False): method get_ref_flows (line 593) | def get_ref_flows(self, x): FILE: code/real/bsrt/model/checkpoint.py function detach_variable (line 5) | def detach_variable(inputs): function check_backward_validity (line 18) | def check_backward_validity(inputs): class CheckpointFunction (line 23) | class CheckpointFunction(torch.autograd.Function): method forward (line 25) | def forward(ctx, run_function, length, *args): method backward (line 34) | def backward(ctx, *output_grads): FILE: code/real/bsrt/model/common.py function default_conv (line 9) | def default_conv(in_channels, out_channels, kernel_size, bias=True): class MeanShift (line 15) | class MeanShift(nn.Conv2d): method __init__ (line 16) | def __init__( class BasicBlock (line 27) | class BasicBlock(nn.Sequential): method __init__ (line 28) | def __init__( class ResBlock (line 41) | class ResBlock(nn.Module): method __init__ (line 42) | def __init__( method forward (line 58) | def forward(self, x): class Upsampler (line 65) | class Upsampler(nn.Sequential): method __init__ (line 66) | def __init__(self, conv, scale, n_feats, bn=False, act=False, bias=True): class UpOnly (line 95) | class UpOnly(nn.Sequential): method __init__ (line 96) | def __init__(self, scale): function lanczos_kernel (line 114) | def lanczos_kernel(dx, a=3, N=None, dtype=None, device=None): function lanczos_shift (line 160) | def lanczos_shift(img, shift, p=5, a=3): FILE: code/real/bsrt/model/non_local/network.py class Network (line 8) | class Network(nn.Module): method __init__ (line 9) | def __init__(self): method forward (line 43) | def forward(self, x): method forward_with_nl_map (line 57) | def forward_with_nl_map(self, x): FILE: code/real/bsrt/model/non_local/non_local_concatenation.py class _NonLocalBlockND (line 6) | class _NonLocalBlockND(nn.Module): method __init__ (line 7) | def __init__(self, in_channels, inter_channels=None, dimension=3, sub_... method forward (line 68) | def forward(self, x, return_nl_map=False): class NONLocalBlock1D (line 109) | class NONLocalBlock1D(_NonLocalBlockND): method __init__ (line 110) | def __init__(self, in_channels, inter_channels=None, sub_sample=True, ... class NONLocalBlock2D (line 117) | class NONLocalBlock2D(_NonLocalBlockND): method __init__ (line 118) | def __init__(self, in_channels, inter_channels=None, sub_sample=True, ... class NONLocalBlock3D (line 125) | class NONLocalBlock3D(_NonLocalBlockND): method __init__ (line 126) | def __init__(self, in_channels, inter_channels=None, sub_sample=True, ... FILE: code/real/bsrt/model/non_local/non_local_cross_dot_product.py class _NonLocalBlockND (line 6) | class _NonLocalBlockND(nn.Module): method __init__ (line 7) | def __init__(self, in_channels, inter_channels=None, dimension=3, sub_... method forward (line 63) | def forward(self, x, ref, return_nl_map=False): class NONLocalBlock1D (line 93) | class NONLocalBlock1D(_NonLocalBlockND): method __init__ (line 94) | def __init__(self, in_channels, inter_channels=None, sub_sample=True, ... class NONLocalBlock2D (line 101) | class NONLocalBlock2D(_NonLocalBlockND): method __init__ (line 102) | def __init__(self, in_channels, inter_channels=None, sub_sample=True, ... class NONLocalBlock3D (line 109) | class NONLocalBlock3D(_NonLocalBlockND): method __init__ (line 110) | def __init__(self, in_channels, inter_channels=None, sub_sample=True, ... FILE: code/real/bsrt/model/non_local/non_local_dot_product.py class _NonLocalBlockND (line 6) | class _NonLocalBlockND(nn.Module): method __init__ (line 7) | def __init__(self, in_channels, inter_channels=None, dimension=3, sub_... method forward (line 63) | def forward(self, x, return_nl_map=False): class NONLocalBlock1D (line 93) | class NONLocalBlock1D(_NonLocalBlockND): method __init__ (line 94) | def __init__(self, in_channels, inter_channels=None, sub_sample=True, ... class NONLocalBlock2D (line 101) | class NONLocalBlock2D(_NonLocalBlockND): method __init__ (line 102) | def __init__(self, in_channels, inter_channels=None, sub_sample=True, ... class NONLocalBlock3D (line 109) | class NONLocalBlock3D(_NonLocalBlockND): method __init__ (line 110) | def __init__(self, in_channels, inter_channels=None, sub_sample=True, ... FILE: code/real/bsrt/model/non_local/non_local_embedded_gaussian.py class _NonLocalBlockND (line 6) | class _NonLocalBlockND(nn.Module): method __init__ (line 7) | def __init__(self, in_channels, inter_channels=None, dimension=3, sub_... method forward (line 70) | def forward(self, x, return_nl_map=False): class NONLocalBlock1D (line 99) | class NONLocalBlock1D(_NonLocalBlockND): method __init__ (line 100) | def __init__(self, in_channels, inter_channels=None, sub_sample=True, ... class NONLocalBlock2D (line 107) | class NONLocalBlock2D(_NonLocalBlockND): method __init__ (line 108) | def __init__(self, in_channels, inter_channels=None, sub_sample=True, ... class NONLocalBlock3D (line 115) | class NONLocalBlock3D(_NonLocalBlockND): method __init__ (line 116) | def __init__(self, in_channels, inter_channels=None, sub_sample=True, ... FILE: code/real/bsrt/model/non_local/non_local_gaussian.py class _NonLocalBlockND (line 6) | class _NonLocalBlockND(nn.Module): method __init__ (line 7) | def __init__(self, in_channels, inter_channels=None, dimension=3, sub_... method forward (line 57) | def forward(self, x, return_nl_map=False): class NONLocalBlock1D (line 95) | class NONLocalBlock1D(_NonLocalBlockND): method __init__ (line 96) | def __init__(self, in_channels, inter_channels=None, sub_sample=True, ... class NONLocalBlock2D (line 103) | class NONLocalBlock2D(_NonLocalBlockND): method __init__ (line 104) | def __init__(self, in_channels, inter_channels=None, sub_sample=True, ... class NONLocalBlock3D (line 111) | class NONLocalBlock3D(_NonLocalBlockND): method __init__ (line 112) | def __init__(self, in_channels, inter_channels=None, sub_sample=True, ... FILE: code/real/bsrt/model/swin_util.py class Mlp (line 17) | class Mlp(nn.Module): method __init__ (line 18) | def __init__(self, in_features, hidden_features=None, out_features=Non... method forward (line 27) | def forward(self, x): class Mlp_GEGLU (line 35) | class Mlp_GEGLU(nn.Module): method __init__ (line 45) | def __init__(self, in_features, hidden_features=None, out_features=Non... method forward (line 56) | def forward(self, x): function window_partition (line 64) | def window_partition(x, window_size): function window_reverse (line 79) | def window_reverse(windows, window_size, H, W): class WindowAttention (line 96) | class WindowAttention(nn.Module): method __init__ (line 110) | def __init__(self, dim, window_size, num_heads, qkv_bias=True, qk_scal... method forward (line 145) | def forward(self, x, mask=None): method extra_repr (line 180) | def extra_repr(self) -> str: method flops (line 183) | def flops(self, N): function calculate_mask (line 197) | def calculate_mask(x_size, window_size, shift_size): class SwinTransformerBlock (line 221) | class SwinTransformerBlock(nn.Module): method __init__ (line 240) | def __init__(self, dim, input_resolution, num_heads, window_size=7, sh... method forward (line 275) | def forward(self, x, x_size): method extra_repr (line 332) | def extra_repr(self) -> str: method flops (line 336) | def flops(self): class PatchMerging (line 351) | class PatchMerging(nn.Module): method __init__ (line 360) | def __init__(self, input_resolution, dim, norm_layer=nn.LayerNorm): method forward (line 367) | def forward(self, x): method extra_repr (line 390) | def extra_repr(self) -> str: method flops (line 393) | def flops(self): class BasicLayer (line 400) | class BasicLayer(nn.Module): method __init__ (line 420) | def __init__(self, dim, input_resolution, depth, num_heads, window_size, method forward (line 448) | def forward(self, x, x_size): method extra_repr (line 459) | def extra_repr(self) -> str: method flops (line 462) | def flops(self): class RSTB (line 471) | class RSTB(nn.Module): method __init__ (line 494) | def __init__(self, dim, input_resolution, depth, num_heads, window_size, method forward (line 536) | def forward(self, x, x_size): method flops (line 541) | def flops(self): class PatchEmbed (line 552) | class PatchEmbed(nn.Module): method __init__ (line 563) | def __init__(self, img_size=224, patch_size=4, in_chans=3, embed_dim=9... method forward (line 581) | def forward(self, x, use_norm=True): method flops (line 587) | def flops(self): class PatchUnEmbed (line 595) | class PatchUnEmbed(nn.Module): method __init__ (line 606) | def __init__(self, img_size=224, patch_size=4, in_chans=3, embed_dim=9... method forward (line 619) | def forward(self, x, x_size): method flops (line 624) | def flops(self): class Upsample (line 629) | class Upsample(nn.Sequential): method __init__ (line 637) | def __init__(self, scale, num_feat): class UpsampleOneStep (line 651) | class UpsampleOneStep(nn.Sequential): method __init__ (line 661) | def __init__(self, scale, num_feat, num_out_ch, input_resolution=None): method flops (line 669) | def flops(self): class SwinIR (line 675) | class SwinIR(nn.Module): method __init__ (line 703) | def __init__(self, img_size=64, patch_size=1, in_chans=3, method _init_weights (line 824) | def _init_weights(self, m): method no_weight_decay (line 834) | def no_weight_decay(self): method no_weight_decay_keywords (line 838) | def no_weight_decay_keywords(self): method check_image_size (line 841) | def check_image_size(self, x): method forward_features (line 848) | def forward_features(self, x): method forward (line 863) | def forward(self, x): method flops (line 899) | def flops(self): FILE: code/real/bsrt/model/utils/interp_methods.py function set_framework_dependencies (line 17) | def set_framework_dependencies(x): function support_sz (line 28) | def support_sz(sz): function cubic (line 35) | def cubic(x): function lanczos2 (line 45) | def lanczos2(x): function lanczos3 (line 51) | def lanczos3(x): function linear (line 57) | def linear(x): function box (line 63) | def box(x): FILE: code/real/bsrt/model/utils/psconv.py class PyConv2d (line 4) | class PyConv2d(nn.Module): method __init__ (line 24) | def __init__(self, in_channels, out_channels, pyconv_kernels, pyconv_g... method forward (line 36) | def forward(self, x): class PSConv2d (line 45) | class PSConv2d(nn.Module): method __init__ (line 46) | def __init__(self, in_channels, out_channels, kernel_size=3, stride=1,... method forward (line 69) | def forward(self, x): class PSGConv2d (line 76) | class PSGConv2d(nn.Module): method __init__ (line 77) | def __init__(self, in_channels, out_channels, kernel_size=3, stride=1,... method forward (line 102) | def forward(self, x): FILE: code/real/bsrt/model/utils/resize_right.py class NoneClass (line 6) | class NoneClass: function resize (line 29) | def resize(input, scale_factors=None, out_shape=None, class ResizeLayer (line 80) | class ResizeLayer(nnModuleWrapped): method __init__ (line 81) | def __init__(self, in_shape, scale_factors=None, out_shape=None, method forward (line 132) | def forward(self, input): function prepare_weights_and_field_of_view_1d (line 147) | def prepare_weights_and_field_of_view_1d(dim, scale_factor, in_sz, out_sz, function apply_weights (line 174) | def apply_weights(input, field_of_view, weights, dim, n_dims, fw): function set_scale_and_out_sz (line 211) | def set_scale_and_out_sz(in_shape, out_shape, scale_factors, fw): function get_projected_grid (line 251) | def get_projected_grid(in_sz, out_sz, scale_factor, fw, device=None): function get_field_of_view (line 266) | def get_field_of_view(projected_grid, cur_support_sz, in_sz, fw, eps): function get_weights (line 288) | def get_weights(interp_method, projected_grid, field_of_view): function apply_antialiasing_if_needed (line 301) | def apply_antialiasing_if_needed(interp_method, support_sz, scale_factor, function fw_ceil (line 315) | def fw_ceil(x, fw): function fw_cat (line 322) | def fw_cat(x, fw): function fw_swapaxes (line 329) | def fw_swapaxes(x, ax_1, ax_2, fw): function fw_set_device (line 335) | def fw_set_device(x, device, fw): FILE: code/real/bsrt/pwcnet/correlation/correlation.py function cupy_kernel (line 236) | def cupy_kernel(strFunction, objVariables): function cupy_launch (line 275) | def cupy_launch(strFunction, strKernel): class _FunctionCorrelation (line 279) | class _FunctionCorrelation(torch.autograd.Function): method forward (line 282) | def forward(self, first, second): method backward (line 336) | def backward(self, gradOutput): function FunctionCorrelation (line 388) | def FunctionCorrelation(tenFirst, tenSecond): class ModuleCorrelation (line 392) | class ModuleCorrelation(torch.nn.Module): method __init__ (line 393) | def __init__(self): method forward (line 397) | def forward(self, tenFirst, tenSecond): FILE: code/real/bsrt/pwcnet/pwcnet.py function backwarp (line 21) | def backwarp(tenInput, tenFlow): class Network (line 44) | class Network(torch.nn.Module): method __init__ (line 45) | def __init__(self): method forward (line 224) | def forward(self, tenFirst, tenSecond): class PWCNet (line 237) | class PWCNet(torch.nn.Module): method __init__ (line 238) | def __init__(self, load_pretrained=True, weights_path=None, rgb2bgr=Fa... method forward (line 253) | def forward(self, source_img, target_img): FILE: code/real/bsrt/pwcnet/run.py function backwarp (line 46) | def backwarp(tenInput, tenFlow): class Network (line 70) | class Network(torch.nn.Module): method __init__ (line 71) | def __init__(self): method forward (line 262) | def forward(self, tenFirst, tenSecond): function estimate (line 280) | def estimate(tenFirst, tenSecond): FILE: code/real/bsrt/scripts/cal_mean_std.py function main (line 9) | def main(): FILE: code/real/bsrt/scripts/download_burstsr_dataset.py function download_burstsr_dataset (line 8) | def download_burstsr_dataset(download_path): function main (line 59) | def main(): FILE: code/real/bsrt/scripts/evaluate_burstsr_val.py class SimpleBaseline (line 8) | class SimpleBaseline: method __init__ (line 9) | def __init__(self): method __call__ (line 12) | def __call__(self, burst): function main (line 19) | def main(): FILE: code/real/bsrt/scripts/save_results_synburst_val.py class SimpleBaseline (line 9) | class SimpleBaseline: method __init__ (line 10) | def __init__(self): method __call__ (line 13) | def __call__(self, burst): function main (line 20) | def main(): FILE: code/real/bsrt/scripts/test_burstsr_dataset.py function main (line 11) | def main(): FILE: code/real/bsrt/scripts/test_synthetic_bursts.py function main (line 11) | def main(): FILE: code/real/bsrt/test.py function main_worker (line 25) | def main_worker(local_rank, nprocs, args): function main (line 58) | def main(): FILE: code/real/bsrt/test_real.py function main (line 31) | def main(): function main_worker (line 35) | def main_worker(local_rank, nprocs, args): FILE: code/real/bsrt/trainer.py class Trainer (line 47) | class Trainer(): method __init__ (line 48) | def __init__(self, args, train_loader, train_sampler, valid_loader, my... method train (line 103) | def train(self): method test (line 210) | def test(self): method save_model (line 313) | def save_model(self, filename): method prepare (line 322) | def prepare(self, *args): method terminate (line 332) | def terminate(self): FILE: code/real/bsrt/utility.py function reduce_mean (line 24) | def reduce_mean(tensor, nprocs): function setup (line 31) | def setup(rank, world_size): function cleanup (line 54) | def cleanup(): function mkdir (line 58) | def mkdir(path): class timer (line 63) | class timer(): method __init__ (line 64) | def __init__(self): method tic (line 68) | def tic(self): method toc (line 71) | def toc(self, restart=False): method hold (line 76) | def hold(self): method release (line 79) | def release(self): method reset (line 85) | def reset(self): class checkpoint (line 89) | class checkpoint(): method __init__ (line 90) | def __init__(self, args): method get_path (line 127) | def get_path(self, *subdir): method save (line 130) | def save(self, trainer, epoch, is_best=False): method add_log (line 139) | def add_log(self, log): method write_log (line 142) | def write_log(self, log, refresh=False): method done (line 149) | def done(self): method plot_psnr (line 152) | def plot_psnr(self, epoch): method begin_background (line 171) | def begin_background(self): method end_background (line 188) | def end_background(self): method save_results (line 193) | def save_results(self, dataset, filename, save_list, scale): function quantize (line 207) | def quantize(img, rgb_range): function calc_psnr (line 212) | def calc_psnr(sr, hr, scale, rgb_range, dataset=None): function make_optimizer (line 231) | def make_optimizer(args, target): function write_gray_to_tfboard (line 287) | def write_gray_to_tfboard(img): function bayer_unify (line 309) | def bayer_unify(raw, input_pattern, target_pattern, mode) -> np.ndarray: function bayer_aug (line 341) | def bayer_aug(raw, flip_h=False, flip_w=False, transpose=False, input_pa... FILE: code/real/bsrt/utils/data_format_utils.py function numpy_to_torch (line 6) | def numpy_to_torch(a: np.ndarray): function torch_to_numpy (line 10) | def torch_to_numpy(a: torch.Tensor): function torch_to_npimage (line 14) | def torch_to_npimage(a: torch.Tensor, unnormalize=True): function npimage_to_torch (line 23) | def npimage_to_torch(a, normalize=True, input_bgr=True): function convert_dict (line 34) | def convert_dict(base_dict, batch_sz): FILE: code/real/bsrt/utils/debayer.py class Debayer3x3 (line 5) | class Debayer3x3(torch.nn.Module): method __init__ (line 29) | def __init__(self): method forward (line 71) | def forward(self, x): class Debayer2x2 (line 91) | class Debayer2x2(torch.nn.Module): method __init__ (line 99) | def __init__(self): method forward (line 115) | def forward(self, x): class DebayerSplit (line 133) | class DebayerSplit(torch.nn.Module): method __init__ (line 140) | def __init__(self): method forward (line 151) | def forward(self, x): FILE: code/real/bsrt/utils/interp_methods.py function set_framework_dependencies (line 17) | def set_framework_dependencies(x): function support_sz (line 28) | def support_sz(sz): function cubic (line 35) | def cubic(x): function lanczos2 (line 45) | def lanczos2(x): function lanczos3 (line 51) | def lanczos3(x): function linear (line 57) | def linear(x): function box (line 63) | def box(x): FILE: code/real/bsrt/utils/metrics.py class MSSSIMLoss (line 15) | class MSSSIMLoss(nn.Module): method __init__ (line 16) | def __init__(self, boundary_ignore=None): method forward (line 21) | def forward(self, pred, gt, valid=None): class CharbonnierLoss (line 33) | class CharbonnierLoss(nn.Module): method __init__ (line 34) | def __init__(self, boundary_ignore=None): method forward (line 39) | def forward(self, pred, gt, valid=None): class L1 (line 51) | class L1(nn.Module): method __init__ (line 52) | def __init__(self, boundary_ignore=None): method forward (line 56) | def forward(self, pred, gt, valid=None): class L2 (line 78) | class L2(nn.Module): method __init__ (line 79) | def __init__(self, boundary_ignore=None): method forward (line 83) | def forward(self, pred, gt, valid=None): class PSNR (line 106) | class PSNR(nn.Module): method __init__ (line 107) | def __init__(self, boundary_ignore=None, max_value=1.0): method psnr (line 112) | def psnr(self, pred, gt, valid=None): method forward (line 119) | def forward(self, pred, gt, valid=None): class AlignedL1 (line 130) | class AlignedL1(nn.Module): method __init__ (line 131) | def __init__(self, alignment_net, sr_factor=4, boundary_ignore=None): method forward (line 139) | def forward(self, pred, gt, burst_input): class AlignedL2 (line 180) | class AlignedL2(nn.Module): method __init__ (line 181) | def __init__(self, alignment_net, sr_factor=4, boundary_ignore=None): method forward (line 190) | def forward(self, pred, gt, burst_input): class AlignedPSNR (line 237) | class AlignedPSNR(nn.Module): method __init__ (line 238) | def __init__(self, alignment_net, sr_factor=4, boundary_ignore=None, m... method psnr (line 243) | def psnr(self, pred, gt, burst_input): method forward (line 250) | def forward(self, pred, gt, burst_input): class AlignedSSIM (line 259) | class AlignedSSIM(nn.Module): method __init__ (line 260) | def __init__(self, alignment_net, sr_factor=4, boundary_ignore=None): method _ssim (line 268) | def _ssim(self, pred, gt, burst_input): method forward (line 307) | def forward(self, pred, gt, burst_input): class AlignedLPIPS (line 313) | class AlignedLPIPS(nn.Module): method __init__ (line 314) | def __init__(self, alignment_net, sr_factor=4, boundary_ignore=None): method _lpips (line 323) | def _lpips(self, pred, gt, burst_input): method forward (line 357) | def forward(self, pred, gt, burst_input): FILE: code/real/bsrt/utils/postprocessing_functions.py class SimplePostProcess (line 7) | class SimplePostProcess: method __init__ (line 8) | def __init__(self, gains=True, ccm=True, gamma=True, smoothstep=True, ... method process (line 15) | def process(self, image, meta_info): function process_linear_image_rgb (line 20) | def process_linear_image_rgb(image, meta_info, gains=True, ccm=True, gam... class BurstSRPostProcess (line 40) | class BurstSRPostProcess: method __init__ (line 41) | def __init__(self, no_white_balance=False, gamma=True, smoothstep=True... method process (line 47) | def process(self, image, meta_info, external_norm_factor=None): function process_burstsr_image_rgb (line 53) | def process_burstsr_image_rgb(im, meta_info, return_np=False, external_n... FILE: code/real/bsrt/utils/resize_right.py class NoneClass (line 6) | class NoneClass: function resize (line 29) | def resize(input, scale_factors=None, out_shape=None, class ResizeLayer (line 80) | class ResizeLayer(nnModuleWrapped): method __init__ (line 81) | def __init__(self, in_shape, scale_factors=None, out_shape=None, method forward (line 132) | def forward(self, input): function prepare_weights_and_field_of_view_1d (line 147) | def prepare_weights_and_field_of_view_1d(dim, scale_factor, in_sz, out_sz, function apply_weights (line 174) | def apply_weights(input, field_of_view, weights, dim, n_dims, fw): function set_scale_and_out_sz (line 211) | def set_scale_and_out_sz(in_shape, out_shape, scale_factors, fw): function get_projected_grid (line 251) | def get_projected_grid(in_sz, out_sz, scale_factor, fw, device=None): function get_field_of_view (line 266) | def get_field_of_view(projected_grid, cur_support_sz, in_sz, fw, eps): function get_weights (line 288) | def get_weights(interp_method, projected_grid, field_of_view): function apply_antialiasing_if_needed (line 301) | def apply_antialiasing_if_needed(interp_method, support_sz, scale_factor, function fw_ceil (line 315) | def fw_ceil(x, fw): function fw_cat (line 322) | def fw_cat(x, fw): function fw_swapaxes (line 329) | def fw_swapaxes(x, ax_1, ax_2, fw): function fw_set_device (line 335) | def fw_set_device(x, device, fw): FILE: code/real/bsrt/utils/spatial_color_alignment.py function gauss_1d (line 6) | def gauss_1d(sz, sigma, center, end_pad=0, density=False): function gauss_2d (line 15) | def gauss_2d(sz, sigma, center, end_pad=(0, 0), density=False): function get_gaussian_kernel (line 29) | def get_gaussian_kernel(sd): function apply_kernel (line 38) | def apply_kernel(im, ksz, gauss_kernel): function match_colors (line 48) | def match_colors(im_ref, im_q, im_test, ksz, gauss_kernel): FILE: code/real/bsrt/utils/stn.py class SpatialTransformer (line 6) | class SpatialTransformer(nn.Module): method __init__ (line 12) | def __init__(self, size, mode='bilinear'): method forward (line 31) | def forward(self, src, flow): FILE: code/real/bsrt/utils/warp.py function warp (line 6) | def warp(feat, flow, mode='bilinear', padding_mode='zeros'): FILE: code/real/bsrt/validate.py function main (line 29) | def main(): function main_worker (line 33) | def main_worker(local_rank, nprocs, args): FILE: code/synthetic/bsrt/data_processing/camera_pipeline.py function random_ccm (line 13) | def random_ccm(): function random_gains (line 47) | def random_gains(): function apply_smoothstep (line 58) | def apply_smoothstep(image): function invert_smoothstep (line 64) | def invert_smoothstep(image): function gamma_expansion (line 70) | def gamma_expansion(image): function gamma_compression (line 76) | def gamma_compression(image): function apply_ccm (line 82) | def apply_ccm(image, ccm): function apply_gains (line 95) | def apply_gains(image, rgb_gain, red_gain, blue_gain): function safe_invert_gains (line 109) | def safe_invert_gains(image, rgb_gain, red_gain, blue_gain): function mosaic (line 125) | def mosaic(image, mode='rggb'): function demosaic (line 151) | def demosaic(image): function random_noise_levels (line 188) | def random_noise_levels(): function add_noise (line 201) | def add_noise(image, shot_noise=0.01, read_noise=0.0005): function process_linear_image_rgb (line 208) | def process_linear_image_rgb(image, meta_info, return_np=False): function process_linear_image_raw (line 225) | def process_linear_image_raw(image, meta_info): FILE: code/synthetic/bsrt/data_processing/synthetic_burst_generation.py function random_crop (line 10) | def random_crop(frames, crop_sz): function rgb2rawburst (line 49) | def rgb2rawburst(image, burst_size, downsample_factor=1, burst_transform... function get_tmat (line 123) | def get_tmat(image_shape, translation, theta, shear_values, scale_factors): function single2lrburst (line 149) | def single2lrburst(image, burst_size, downsample_factor=1, transformatio... FILE: code/synthetic/bsrt/datasets/burstsr_dataset.py class SamsungRAWImage (line 10) | class SamsungRAWImage: method load (line 12) | def load(path): method __init__ (line 24) | def __init__(self, im_raw, black_level, cam_wb, daylight_wb, color_mat... method get_all_meta_data (line 38) | def get_all_meta_data(self): method get_exposure_time (line 42) | def get_exposure_time(self): method get_noise_profile (line 45) | def get_noise_profile(self): method get_f_number (line 51) | def get_f_number(self): method get_iso (line 54) | def get_iso(self): method get_image_data (line 57) | def get_image_data(self, substract_black_level=False, white_balance=Fa... method shape (line 72) | def shape(self): method crop_image (line 76) | def crop_image(self, r1, r2, c1, c2): method get_crop (line 79) | def get_crop(self, r1, r2, c1, c2): method postprocess (line 90) | def postprocess(self, return_np=True, norm_factor=None): class CanonImage (line 113) | class CanonImage: method load (line 115) | def load(path, split='train'): method __init__ (line 125) | def __init__(self, im_raw, black_level, cam_wb, daylight_wb, rgb_xyz_m... method shape (line 150) | def shape(self): method get_all_meta_data (line 154) | def get_all_meta_data(self): method get_exposure_time (line 159) | def get_exposure_time(self): method get_f_number (line 162) | def get_f_number(self): method get_iso (line 165) | def get_iso(self): method get_image_data (line 168) | def get_image_data(self, substract_black_level=False, white_balance=Fa... method set_image_data (line 182) | def set_image_data(self, im_data): method crop_image (line 185) | def crop_image(self, r1, r2, c1, c2): method get_crop (line 188) | def get_crop(self, r1, r2, c1, c2): method set_crop_info (line 193) | def set_crop_info(self, crop_info): method resize (line 196) | def resize(self, size=None, scale_factor=None): method postprocess (line 201) | def postprocess(self, return_np=True): function load_txt (line 216) | def load_txt(path): class BurstSRDataset (line 223) | class BurstSRDataset(torch.utils.data.Dataset): method __init__ (line 225) | def __init__(self, root, burst_size=8, crop_sz=80, center_crop=False, ... method _get_burst_list (line 255) | def _get_burst_list(self): method get_burst_info (line 260) | def get_burst_info(self, burst_id): method _get_raw_image (line 264) | def _get_raw_image(self, burst_id, im_id): method _get_gt_image (line 268) | def _get_gt_image(self, burst_id): method get_burst (line 272) | def get_burst(self, burst_id, im_ids, info=None): method _sample_images (line 281) | def _sample_images(self): method __len__ (line 288) | def __len__(self): method __getitem__ (line 291) | def __getitem__(self, index): function pack_raw_image (line 400) | def pack_raw_image(im_raw): function flatten_raw_image (line 415) | def flatten_raw_image(im_raw_4ch): function pack_raw_image_batch (line 430) | def pack_raw_image_batch(im_raw): function flatten_raw_image_batch (line 439) | def flatten_raw_image_batch(im_raw_4ch): FILE: code/synthetic/bsrt/datasets/burstsr_test_dataset.py class BurstSRDataset (line 8) | class BurstSRDataset(torch.utils.data.Dataset): method __init__ (line 10) | def __init__(self, root, burst_size=8, crop_sz=80, center_crop=False, ... method _get_burst_list (line 40) | def _get_burst_list(self): method get_burst_info (line 45) | def get_burst_info(self, burst_id): method _get_raw_image (line 49) | def _get_raw_image(self, burst_id, im_id): method get_burst (line 53) | def get_burst(self, burst_id, im_ids, info=None): method _sample_images (line 61) | def _sample_images(self): method __len__ (line 68) | def __len__(self): method __getitem__ (line 71) | def __getitem__(self, index): FILE: code/synthetic/bsrt/datasets/data_sampler.py class DistIterSampler (line 13) | class DistIterSampler(Sampler): method __init__ (line 31) | def __init__(self, dataset, num_replicas=None, rank=None, ratio=100): method __iter__ (line 47) | def __iter__(self): method __len__ (line 64) | def __len__(self): method set_epoch (line 67) | def set_epoch(self, epoch): FILE: code/synthetic/bsrt/datasets/realworld_burst_test_set.py class RealWorldBurstTest (line 7) | class RealWorldBurstTest(torch.utils.data.Dataset): method __init__ (line 10) | def __init__(self, root): method __len__ (line 15) | def __len__(self): method _read_burst_image (line 18) | def _read_burst_image(self, index, image_id): method __getitem__ (line 23) | def __getitem__(self, index): FILE: code/synthetic/bsrt/datasets/synthetic_burst_test_set.py class SyntheticBurstTest (line 7) | class SyntheticBurstTest(torch.utils.data.Dataset): method __init__ (line 11) | def __init__(self, root): method __len__ (line 16) | def __len__(self): method _read_burst_image (line 19) | def _read_burst_image(self, index, image_id): method __getitem__ (line 24) | def __getitem__(self, index): FILE: code/synthetic/bsrt/datasets/synthetic_burst_train_set.py class SyntheticBurst (line 8) | class SyntheticBurst(torch.utils.data.Dataset): method __init__ (line 18) | def __init__(self, base_dataset, burst_size=8, crop_sz=384, transform=... method __len__ (line 37) | def __len__(self): method __getitem__ (line 40) | def __getitem__(self, index): FILE: code/synthetic/bsrt/datasets/synthetic_burst_val_set.py class SyntheticBurstVal (line 8) | class SyntheticBurstVal(torch.utils.data.Dataset): method __init__ (line 15) | def __init__(self, root=None, initialize=True): method initialize (line 25) | def initialize(self): method __len__ (line 28) | def __len__(self): method _read_burst_image (line 31) | def _read_burst_image(self, index, image_id): method _read_gt_image (line 37) | def _read_gt_image(self, index): method _read_meta_info (line 42) | def _read_meta_info(self, index): method __getitem__ (line 48) | def __getitem__(self, index): FILE: code/synthetic/bsrt/datasets/zurich_raw2rgb_dataset.py class ZurichRAW2RGB (line 7) | class ZurichRAW2RGB(torch.utils.data.Dataset): method __init__ (line 12) | def __init__(self, root, split='train'): method _get_image_list (line 22) | def _get_image_list(self, split): method _get_image (line 33) | def _get_image(self, im_id): method get_image (line 38) | def get_image(self, im_id): method __len__ (line 43) | def __len__(self): method __getitem__ (line 46) | def __getitem__(self, index): FILE: code/synthetic/bsrt/loss/Charbonnier.py class CharbonnierLoss (line 5) | class CharbonnierLoss(nn.Module): method __init__ (line 8) | def __init__(self, epsilon=1e-3, reduce=True): method forward (line 13) | def forward(self, X, Y): FILE: code/synthetic/bsrt/loss/__init__.py class Loss (line 14) | class Loss(nn.modules.loss._Loss): method __init__ (line 15) | def __init__(self, args, ckp): method forward (line 80) | def forward(self, sr, hr): method step (line 97) | def step(self): method start_log (line 102) | def start_log(self): method end_log (line 105) | def end_log(self, n_batches): method display_loss (line 108) | def display_loss(self, batch): method plot_loss (line 116) | def plot_loss(self, apath, epoch): method get_loss_module (line 130) | def get_loss_module(self): method save (line 136) | def save(self, apath): method load (line 140) | def load(self, apath, cpu=False): FILE: code/synthetic/bsrt/loss/adversarial.py class Adversarial (line 12) | class Adversarial(nn.Module): method __init__ (line 13) | def __init__(self, args, gan_type): method forward (line 36) | def forward(self, fake, real): method state_dict (line 96) | def state_dict(self, *args, **kwargs): method bce (line 102) | def bce(self, real, fake): FILE: code/synthetic/bsrt/loss/discriminator.py class Discriminator (line 5) | class Discriminator(nn.Module): method __init__ (line 9) | def __init__(self, args, gan_type='GAN'): method forward (line 65) | def forward(self, x): FILE: code/synthetic/bsrt/loss/filter.py class Filter (line 5) | class Filter(nn.Module): method __init__ (line 6) | def __init__(self, args): method forward (line 16) | def forward(self, x, y): FILE: code/synthetic/bsrt/loss/hist_entropy.py class HistEntropy (line 5) | class HistEntropy(nn.Module): method __init__ (line 6) | def __init__(self, args): method forward (line 10) | def forward(self, x): FILE: code/synthetic/bsrt/loss/mssim.py function gaussian (line 7) | def gaussian(window_size, sigma): function create_window (line 12) | def create_window(window_size, channel=1): function ssim (line 19) | def ssim(img1, img2, window_size=11, window=None, size_average=True, ful... function msssim (line 71) | def msssim(img1, img2, window_size=11, size_average=True, val_range=None... class SSIM (line 109) | class SSIM(torch.nn.Module): method __init__ (line 110) | def __init__(self, window_size=11, size_average=True, val_range=None): method forward (line 120) | def forward(self, img1, img2): class MSSSIM (line 132) | class MSSSIM(torch.nn.Module): method __init__ (line 133) | def __init__(self, window_size=11, size_average=True, channel=3): method forward (line 139) | def forward(self, img1, img2): FILE: code/synthetic/bsrt/loss/vgg.py class VGG (line 8) | class VGG(nn.Module): method __init__ (line 9) | def __init__(self, conv_index, rgb_range=1): method forward (line 24) | def forward(self, sr, hr): FILE: code/synthetic/bsrt/main.py function init_seeds (line 23) | def init_seeds(seed=0, cuda_deterministic=True): function main (line 38) | def main(): function main_worker (line 45) | def main_worker(local_rank, nprocs, args): FILE: code/synthetic/bsrt/model/DCNv2/dcn_v2.py class _DCNv2 (line 17) | class _DCNv2(Function): method forward (line 21) | def forward( method backward (line 52) | def backward(ctx, grad_output): method symbolic (line 75) | def symbolic( class DCNv2 (line 102) | class DCNv2(nn.Module): method __init__ (line 103) | def __init__( method reset_parameters (line 126) | def reset_parameters(self): method forward (line 134) | def forward(self, input, offset, mask): class DCN (line 153) | class DCN(DCNv2): method __init__ (line 154) | def __init__( method init_offset (line 179) | def init_offset(self): method forward (line 183) | def forward(self, input): class DCN_sep (line 201) | class DCN_sep(DCNv2): method __init__ (line 204) | def __init__(self, method init_offset (line 225) | def init_offset(self): method forward (line 229) | def forward(self, input, fea): class FlowGuidedDCN (line 249) | class FlowGuidedDCN(DCNv2): method __init__ (line 252) | def __init__(self, method init_offset (line 269) | def init_offset(self): method forward (line 273) | def forward(self, input, fea, flows): class InsideFlowGuidedDCN (line 296) | class InsideFlowGuidedDCN(DCNv2): method __init__ (line 299) | def __init__(self, method reset_parameters (line 322) | def reset_parameters(self): method init_offset (line 331) | def init_offset(self): method forward (line 335) | def forward(self, input, warped, ref, flows): class _DCNv2Pooling (line 359) | class _DCNv2Pooling(Function): method forward (line 361) | def forward( method backward (line 402) | def backward(ctx, grad_output): class DCNv2Pooling (line 426) | class DCNv2Pooling(nn.Module): method __init__ (line 427) | def __init__( method forward (line 448) | def forward(self, input, rois, offset): class DCNPooling (line 467) | class DCNPooling(DCNv2Pooling): method __init__ (line 468) | def __init__( method forward (line 506) | def forward(self, input, rois): FILE: code/synthetic/bsrt/model/DCNv2/setup.py function get_extensions (line 13) | def get_extensions(): FILE: code/synthetic/bsrt/model/DCNv2/src/cpu/dcn_v2_cpu.cpp function dcn_v2_cpu_forward (line 17) | at::Tensor function dcn_v2_cpu_backward (line 109) | std::vector dcn_v2_cpu_backward(const at::Tensor &input, FILE: code/synthetic/bsrt/model/DCNv2/src/cpu/dcn_v2_im2col_cpu.cpp function dmcn_im2col_bilinear_cpu (line 27) | float dmcn_im2col_bilinear_cpu(const float *bottom_data, const int data_... function dmcn_get_gradient_weight_cpu (line 58) | float dmcn_get_gradient_weight_cpu(float argmax_h, float argmax_w, function dmcn_get_coordinate_weight_cpu (line 84) | float dmcn_get_coordinate_weight_cpu(float argmax_h, float argmax_w, function modulated_deformable_im2col_cpu_kernel (line 127) | void modulated_deformable_im2col_cpu_kernel(const int n, const float *da... function modulated_deformable_col2im_cpu_kernel (line 198) | void modulated_deformable_col2im_cpu_kernel(const int n, const float *da... function modulated_deformable_col2im_coord_cpu_kernel (line 259) | void modulated_deformable_col2im_coord_cpu_kernel(const int n, const flo... function modulated_deformable_im2col_cpu (line 331) | void modulated_deformable_im2col_cpu(const float* data_im, const float* ... function modulated_deformable_col2im_cpu (line 353) | void modulated_deformable_col2im_cpu(const float* data_col, const float*... function modulated_deformable_col2im_coord_cpu (line 375) | void modulated_deformable_col2im_coord_cpu(const float* data_col, const ... FILE: code/synthetic/bsrt/model/DCNv2/src/cpu/dcn_v2_psroi_pooling_cpu.cpp function T (line 34) | T bilinear_interp_cpu( function DeformablePSROIPoolForwardKernelCpu (line 59) | void DeformablePSROIPoolForwardKernelCpu( function DeformablePSROIPoolBackwardAccKernelCpu (line 149) | void DeformablePSROIPoolBackwardAccKernelCpu( function dcn_v2_psroi_pooling_cpu_forward (line 278) | std::tuple function dcn_v2_psroi_pooling_cpu_backward (line 350) | std::tuple FILE: code/synthetic/bsrt/model/DCNv2/src/vision.cpp function PYBIND11_MODULE (line 4) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) { FILE: code/synthetic/bsrt/model/DCNv2/test.py function conv_identify (line 20) | def conv_identify(weight, bias): function check_zero_offset (line 32) | def check_zero_offset(): function check_gradient_dconv (line 69) | def check_gradient_dconv(): function check_pooling_zero_offset (line 100) | def check_pooling_zero_offset(): function check_gradient_dpooling (line 134) | def check_gradient_dpooling(): function example_dconv (line 169) | def example_dconv(): function example_dpooling (line 183) | def example_dpooling(): function example_mdpooling (line 226) | def example_mdpooling(): FILE: code/synthetic/bsrt/model/__init__.py class Model (line 10) | class Model(nn.Module): method __init__ (line 11) | def __init__(self, args, ckp): method forward (line 53) | def forward(self, x, idx_scale): method save (line 73) | def save(self, apath, epoch, is_best=False): method load (line 90) | def load(self, apath, pre_train='', resume=-1, cpu=False): method forward_chop (line 217) | def forward_chop(self, *args, shave=10, min_size=160000): method forward_x8 (line 276) | def forward_x8(self, *args, forward_function=None): FILE: code/synthetic/bsrt/model/arch_util.py function initialize_weights (line 9) | def initialize_weights(net_l, scale=1): function make_layer (line 29) | def make_layer(block, n_layers): function conv_layer (line 38) | def conv_layer(in_channels, out_channels, kernel_size, stride=1, padding... class ESA (line 42) | class ESA(nn.Module): method __init__ (line 43) | def __init__(self, n_feats, conv=conv_layer): method forward (line 56) | def forward(self, x): class DWConv (line 71) | class DWConv(nn.Module): method __init__ (line 72) | def __init__(self, dim=768): method forward (line 76) | def forward(self, x): class SELayer (line 82) | class SELayer(nn.Module): method __init__ (line 86) | def __init__(self, channel, reduction=16): method forward (line 96) | def forward(self, x): class ResidualBlock_noBN (line 102) | class ResidualBlock_noBN(nn.Module): method __init__ (line 108) | def __init__(self, nf=64): method forward (line 116) | def forward(self, x): class ResidualBlock_SE (line 123) | class ResidualBlock_SE(nn.Module): method __init__ (line 129) | def __init__(self, nf=64, reduction=16): method forward (line 138) | def forward(self, x): class _PositionAttentionModule (line 148) | class _PositionAttentionModule(nn.Module): method __init__ (line 151) | def __init__(self, in_channels, **kwargs): method forward (line 159) | def forward(self, x): class SALayer (line 171) | class SALayer(nn.Module): method __init__ (line 172) | def __init__(self, wn=None): method forward (line 178) | def forward(self, x): class CALayerV2 (line 186) | class CALayerV2(nn.Module): method __init__ (line 187) | def __init__(self, n_feat, reduction=16, wn=None): method forward (line 200) | def forward(self, x): class DALayer (line 207) | class DALayer(nn.Module): method __init__ (line 208) | def __init__(self, channel, reduction, wn): method forward (line 215) | def forward(self, x): class CALayer (line 223) | class CALayer(nn.Module): method __init__ (line 224) | def __init__(self, channel, reduction, wn): method forward (line 236) | def forward(self, x): class RCAB (line 243) | class RCAB(nn.Module): method __init__ (line 244) | def __init__( method forward (line 266) | def forward(self, x): class ResidualGroup (line 273) | class ResidualGroup(nn.Module): method __init__ (line 274) | def __init__(self, n_feat, n_resblocks, da=False): method forward (line 292) | def forward(self, x): function make_layer_idx (line 302) | def make_layer_idx(block, n_layers): class LRSCRCAB (line 309) | class LRSCRCAB(nn.Module): method __init__ (line 310) | def __init__( method forward (line 332) | def forward(self, x): class LRSCPYRCAB (line 339) | class LRSCPYRCAB(nn.Module): method __init__ (line 340) | def __init__( method forward (line 366) | def forward(self, x): class LRSCResidualGroup (line 372) | class LRSCResidualGroup(nn.Module): method __init__ (line 373) | def __init__(self, n_feat, n_resblocks, da=False, idx=0): method forward (line 392) | def forward(self, x): class LRSCPSResidualGroup (line 400) | class LRSCPSResidualGroup(nn.Module): method __init__ (line 401) | def __init__(self, n_feat, n_resblocks, da=False, idx=0): method forward (line 421) | def forward(self, x): class LRSCPyResidualGroup (line 430) | class LRSCPyResidualGroup(nn.Module): method __init__ (line 431) | def __init__(self, n_feat, n_resblocks, da=False, idx=0): method forward (line 451) | def forward(self, x): class LRSCWideActResBlock (line 458) | class LRSCWideActResBlock(nn.Module): method __init__ (line 459) | def __init__(self, nf=64, idx=0): method forward (line 482) | def forward(self, x): class LRSCPyWideActResBlock (line 488) | class LRSCPyWideActResBlock(nn.Module): method __init__ (line 489) | def __init__(self, nf=64, idx=0): method forward (line 515) | def forward(self, x): class LRSCPyWideActResGroup (line 523) | class LRSCPyWideActResGroup(nn.Module): method __init__ (line 524) | def __init__(self, nf, n_resblocks, idx=0): method forward (line 539) | def forward(self, x): class LRSCWideActResGroup (line 548) | class LRSCWideActResGroup(nn.Module): method __init__ (line 549) | def __init__(self, nf, n_resblocks, idx=0): method forward (line 564) | def forward(self, x): class PYRCAB (line 577) | class PYRCAB(nn.Module): method __init__ (line 578) | def __init__( method forward (line 603) | def forward(self, x): class PyResidualGroup (line 609) | class PyResidualGroup(nn.Module): method __init__ (line 610) | def __init__(self, n_feat, n_resblocks, da=False): method forward (line 632) | def forward(self, x): class WideActResBlock (line 637) | class WideActResBlock(nn.Module): method __init__ (line 638) | def __init__(self, nf=64): method forward (line 658) | def forward(self, x): class PSWideActResBlock (line 664) | class PSWideActResBlock(nn.Module): method __init__ (line 665) | def __init__(self, nf=64): method forward (line 685) | def forward(self, x): class PyWideActResBlock (line 691) | class PyWideActResBlock(nn.Module): method __init__ (line 692) | def __init__(self, nf=64): method forward (line 717) | def forward(self, x): function flow_warp (line 723) | def flow_warp(x, flow, interp_mode='bilinear', padding_mode='zeros', ali... FILE: code/synthetic/bsrt/model/bsrt.py function make_model (line 27) | def make_model(args, parent=False): class BasicModule (line 66) | class BasicModule(nn.Module): method __init__ (line 70) | def __init__(self): method forward (line 80) | def forward(self, tensor_input): class SpyNet (line 84) | class SpyNet(nn.Module): method __init__ (line 92) | def __init__(self, load_path=None, return_levels=[5]): method preprocess (line 110) | def preprocess(self, tensor_input): method process (line 114) | def process(self, ref, supp, w, h, w_floor, h_floor): method forward (line 160) | def forward(self, ref, supp): class FlowGuidedPCDAlign (line 176) | class FlowGuidedPCDAlign(nn.Module): method __init__ (line 181) | def __init__(self, nf=64, groups=8): method forward (line 209) | def forward(self, nbr_fea_l, nbr_fea_warped_l, ref_fea_l, flows_l): class CrossNonLocal_Fusion (line 246) | class CrossNonLocal_Fusion(nn.Module): method __init__ (line 249) | def __init__(self, nf=64, out_feat=96, nframes=5, center=2): method forward (line 265) | def forward(self, aligned_fea): class BSRT (line 287) | class BSRT(nn.Module): method __init__ (line 288) | def __init__(self, args, nframes=8, img_size=64, patch_size=1, in_chan... method _init_weights (line 452) | def _init_weights(self, m): method no_weight_decay (line 462) | def no_weight_decay(self): method no_weight_decay_keywords (line 466) | def no_weight_decay_keywords(self): method _upsample_add (line 469) | def _upsample_add(self, x, y): method check_image_size (line 472) | def check_image_size(self, x): method pre_forward_features (line 479) | def pre_forward_features(self, x): method forward_features (line 498) | def forward_features(self, x): method forward (line 516) | def forward(self, x, print_time=False): method get_ref_flows (line 593) | def get_ref_flows(self, x): FILE: code/synthetic/bsrt/model/checkpoint.py function detach_variable (line 5) | def detach_variable(inputs): function check_backward_validity (line 18) | def check_backward_validity(inputs): class CheckpointFunction (line 23) | class CheckpointFunction(torch.autograd.Function): method forward (line 25) | def forward(ctx, run_function, length, *args): method backward (line 34) | def backward(ctx, *output_grads): FILE: code/synthetic/bsrt/model/common.py function default_conv (line 9) | def default_conv(in_channels, out_channels, kernel_size, bias=True): class MeanShift (line 15) | class MeanShift(nn.Conv2d): method __init__ (line 16) | def __init__( class BasicBlock (line 27) | class BasicBlock(nn.Sequential): method __init__ (line 28) | def __init__( class ResBlock (line 41) | class ResBlock(nn.Module): method __init__ (line 42) | def __init__( method forward (line 58) | def forward(self, x): class Upsampler (line 65) | class Upsampler(nn.Sequential): method __init__ (line 66) | def __init__(self, conv, scale, n_feats, bn=False, act=False, bias=True): class UpOnly (line 95) | class UpOnly(nn.Sequential): method __init__ (line 96) | def __init__(self, scale): function lanczos_kernel (line 114) | def lanczos_kernel(dx, a=3, N=None, dtype=None, device=None): function lanczos_shift (line 160) | def lanczos_shift(img, shift, p=5, a=3): FILE: code/synthetic/bsrt/model/ebsr.py function make_model (line 27) | def make_model(args, parent=False): class BasicModule (line 58) | class BasicModule(nn.Module): method __init__ (line 62) | def __init__(self): method forward (line 72) | def forward(self, tensor_input): class SpyNet (line 76) | class SpyNet(nn.Module): method __init__ (line 84) | def __init__(self, load_path=None, return_levels=[5]): method preprocess (line 102) | def preprocess(self, tensor_input): method process (line 106) | def process(self, ref, supp, w, h, w_floor, h_floor): method forward (line 152) | def forward(self, ref, supp): class PCD_Align (line 167) | class PCD_Align(nn.Module): method __init__ (line 172) | def __init__(self, nf=64, groups=8, wn=None): method forward (line 206) | def forward(self, nbr_fea_l, ref_fea_l): class FlowGuidedPCDAlign (line 245) | class FlowGuidedPCDAlign(nn.Module): method __init__ (line 250) | def __init__(self, nf=64, groups=8): method forward (line 277) | def forward(self, nbr_fea_l, nbr_fea_warped_l, ref_fea_l, flows_l): class CrossNonLocal_Fusion (line 315) | class CrossNonLocal_Fusion(nn.Module): method __init__ (line 318) | def __init__(self, nf=64, out_feat=96, nframes=5, center=2): method forward (line 334) | def forward(self, aligned_fea): class EBSR (line 356) | class EBSR(nn.Module): method __init__ (line 360) | def __init__(self, args, nframes=8, img_size=64, patch_size=1, in_chan... method _init_weights (line 536) | def _init_weights(self, m): method no_weight_decay (line 546) | def no_weight_decay(self): method no_weight_decay_keywords (line 550) | def no_weight_decay_keywords(self): method _upsample_add (line 553) | def _upsample_add(self, x, y): method check_image_size (line 556) | def check_image_size(self, x): method pre_forward_features (line 563) | def pre_forward_features(self, x): method forward_features (line 582) | def forward_features(self, x): method forward (line 600) | def forward(self, x, print_time=False): method get_ref_flows (line 679) | def get_ref_flows(self, x): method get_flow_2frames (line 694) | def get_flow_2frames(self, x): method get_aligned_image_2frames (line 714) | def get_aligned_image_2frames(self, x, flows_backward, flows_forward): method get_aligned_feature_2frames (line 735) | def get_aligned_feature_2frames(self, x): FILE: code/synthetic/bsrt/model/non_local/network.py class Network (line 8) | class Network(nn.Module): method __init__ (line 9) | def __init__(self): method forward (line 43) | def forward(self, x): method forward_with_nl_map (line 57) | def forward_with_nl_map(self, x): FILE: code/synthetic/bsrt/model/non_local/non_local_concatenation.py class _NonLocalBlockND (line 6) | class _NonLocalBlockND(nn.Module): method __init__ (line 7) | def __init__(self, in_channels, inter_channels=None, dimension=3, sub_... method forward (line 68) | def forward(self, x, return_nl_map=False): class NONLocalBlock1D (line 109) | class NONLocalBlock1D(_NonLocalBlockND): method __init__ (line 110) | def __init__(self, in_channels, inter_channels=None, sub_sample=True, ... class NONLocalBlock2D (line 117) | class NONLocalBlock2D(_NonLocalBlockND): method __init__ (line 118) | def __init__(self, in_channels, inter_channels=None, sub_sample=True, ... class NONLocalBlock3D (line 125) | class NONLocalBlock3D(_NonLocalBlockND): method __init__ (line 126) | def __init__(self, in_channels, inter_channels=None, sub_sample=True, ... FILE: code/synthetic/bsrt/model/non_local/non_local_cross_dot_product.py class _NonLocalBlockND (line 6) | class _NonLocalBlockND(nn.Module): method __init__ (line 7) | def __init__(self, in_channels, inter_channels=None, dimension=3, sub_... method forward (line 63) | def forward(self, x, ref, return_nl_map=False): class NONLocalBlock1D (line 93) | class NONLocalBlock1D(_NonLocalBlockND): method __init__ (line 94) | def __init__(self, in_channels, inter_channels=None, sub_sample=True, ... class NONLocalBlock2D (line 101) | class NONLocalBlock2D(_NonLocalBlockND): method __init__ (line 102) | def __init__(self, in_channels, inter_channels=None, sub_sample=True, ... class NONLocalBlock3D (line 109) | class NONLocalBlock3D(_NonLocalBlockND): method __init__ (line 110) | def __init__(self, in_channels, inter_channels=None, sub_sample=True, ... FILE: code/synthetic/bsrt/model/non_local/non_local_dot_product.py class _NonLocalBlockND (line 6) | class _NonLocalBlockND(nn.Module): method __init__ (line 7) | def __init__(self, in_channels, inter_channels=None, dimension=3, sub_... method forward (line 63) | def forward(self, x, return_nl_map=False): class NONLocalBlock1D (line 93) | class NONLocalBlock1D(_NonLocalBlockND): method __init__ (line 94) | def __init__(self, in_channels, inter_channels=None, sub_sample=True, ... class NONLocalBlock2D (line 101) | class NONLocalBlock2D(_NonLocalBlockND): method __init__ (line 102) | def __init__(self, in_channels, inter_channels=None, sub_sample=True, ... class NONLocalBlock3D (line 109) | class NONLocalBlock3D(_NonLocalBlockND): method __init__ (line 110) | def __init__(self, in_channels, inter_channels=None, sub_sample=True, ... FILE: code/synthetic/bsrt/model/non_local/non_local_embedded_gaussian.py class _NonLocalBlockND (line 6) | class _NonLocalBlockND(nn.Module): method __init__ (line 7) | def __init__(self, in_channels, inter_channels=None, dimension=3, sub_... method forward (line 70) | def forward(self, x, return_nl_map=False): class NONLocalBlock1D (line 99) | class NONLocalBlock1D(_NonLocalBlockND): method __init__ (line 100) | def __init__(self, in_channels, inter_channels=None, sub_sample=True, ... class NONLocalBlock2D (line 107) | class NONLocalBlock2D(_NonLocalBlockND): method __init__ (line 108) | def __init__(self, in_channels, inter_channels=None, sub_sample=True, ... class NONLocalBlock3D (line 115) | class NONLocalBlock3D(_NonLocalBlockND): method __init__ (line 116) | def __init__(self, in_channels, inter_channels=None, sub_sample=True, ... FILE: code/synthetic/bsrt/model/non_local/non_local_gaussian.py class _NonLocalBlockND (line 6) | class _NonLocalBlockND(nn.Module): method __init__ (line 7) | def __init__(self, in_channels, inter_channels=None, dimension=3, sub_... method forward (line 57) | def forward(self, x, return_nl_map=False): class NONLocalBlock1D (line 95) | class NONLocalBlock1D(_NonLocalBlockND): method __init__ (line 96) | def __init__(self, in_channels, inter_channels=None, sub_sample=True, ... class NONLocalBlock2D (line 103) | class NONLocalBlock2D(_NonLocalBlockND): method __init__ (line 104) | def __init__(self, in_channels, inter_channels=None, sub_sample=True, ... class NONLocalBlock3D (line 111) | class NONLocalBlock3D(_NonLocalBlockND): method __init__ (line 112) | def __init__(self, in_channels, inter_channels=None, sub_sample=True, ... FILE: code/synthetic/bsrt/model/swin_util.py class Mlp (line 17) | class Mlp(nn.Module): method __init__ (line 18) | def __init__(self, in_features, hidden_features=None, out_features=Non... method forward (line 27) | def forward(self, x): function window_partition (line 36) | def window_partition(x, window_size): function window_reverse (line 51) | def window_reverse(windows, window_size, H, W): class WindowAttention (line 68) | class WindowAttention(nn.Module): method __init__ (line 82) | def __init__(self, dim, window_size, num_heads, qkv_bias=True, qk_scal... method forward (line 117) | def forward(self, x, mask=None): method extra_repr (line 152) | def extra_repr(self) -> str: method flops (line 155) | def flops(self, N): function calculate_mask (line 169) | def calculate_mask(x_size, window_size, shift_size): class SwinTransformerBlock (line 193) | class SwinTransformerBlock(nn.Module): method __init__ (line 212) | def __init__(self, dim, input_resolution, num_heads, window_size=7, sh... method forward (line 240) | def forward(self, x, x_size): method extra_repr (line 281) | def extra_repr(self) -> str: method flops (line 285) | def flops(self): class PatchMerging (line 300) | class PatchMerging(nn.Module): method __init__ (line 309) | def __init__(self, input_resolution, dim, norm_layer=nn.LayerNorm): method forward (line 316) | def forward(self, x): method extra_repr (line 339) | def extra_repr(self) -> str: method flops (line 342) | def flops(self): class BasicLayer (line 349) | class BasicLayer(nn.Module): method __init__ (line 369) | def __init__(self, dim, input_resolution, depth, num_heads, window_size, method forward (line 397) | def forward(self, x, x_size): method extra_repr (line 408) | def extra_repr(self) -> str: method flops (line 411) | def flops(self): class RSTB (line 420) | class RSTB(nn.Module): method __init__ (line 443) | def __init__(self, dim, input_resolution, depth, num_heads, window_size, method forward (line 485) | def forward(self, x, x_size): method flops (line 490) | def flops(self): class PatchEmbed (line 501) | class PatchEmbed(nn.Module): method __init__ (line 512) | def __init__(self, img_size=224, patch_size=4, in_chans=3, embed_dim=9... method forward (line 530) | def forward(self, x, use_norm=True): method flops (line 536) | def flops(self): class PatchUnEmbed (line 544) | class PatchUnEmbed(nn.Module): method __init__ (line 555) | def __init__(self, img_size=224, patch_size=4, in_chans=3, embed_dim=9... method forward (line 568) | def forward(self, x, x_size): method flops (line 573) | def flops(self): FILE: code/synthetic/bsrt/model/utils/interp_methods.py function set_framework_dependencies (line 17) | def set_framework_dependencies(x): function support_sz (line 28) | def support_sz(sz): function cubic (line 35) | def cubic(x): function lanczos2 (line 45) | def lanczos2(x): function lanczos3 (line 51) | def lanczos3(x): function linear (line 57) | def linear(x): function box (line 63) | def box(x): FILE: code/synthetic/bsrt/model/utils/psconv.py class PyConv2d (line 4) | class PyConv2d(nn.Module): method __init__ (line 24) | def __init__(self, in_channels, out_channels, pyconv_kernels, pyconv_g... method forward (line 36) | def forward(self, x): class PSConv2d (line 45) | class PSConv2d(nn.Module): method __init__ (line 46) | def __init__(self, in_channels, out_channels, kernel_size=3, stride=1,... method forward (line 69) | def forward(self, x): class PSGConv2d (line 76) | class PSGConv2d(nn.Module): method __init__ (line 77) | def __init__(self, in_channels, out_channels, kernel_size=3, stride=1,... method forward (line 102) | def forward(self, x): FILE: code/synthetic/bsrt/model/utils/resize_right.py class NoneClass (line 6) | class NoneClass: function resize (line 29) | def resize(input, scale_factors=None, out_shape=None, class ResizeLayer (line 80) | class ResizeLayer(nnModuleWrapped): method __init__ (line 81) | def __init__(self, in_shape, scale_factors=None, out_shape=None, method forward (line 132) | def forward(self, input): function prepare_weights_and_field_of_view_1d (line 147) | def prepare_weights_and_field_of_view_1d(dim, scale_factor, in_sz, out_sz, function apply_weights (line 174) | def apply_weights(input, field_of_view, weights, dim, n_dims, fw): function set_scale_and_out_sz (line 211) | def set_scale_and_out_sz(in_shape, out_shape, scale_factors, fw): function get_projected_grid (line 251) | def get_projected_grid(in_sz, out_sz, scale_factor, fw, device=None): function get_field_of_view (line 266) | def get_field_of_view(projected_grid, cur_support_sz, in_sz, fw, eps): function get_weights (line 288) | def get_weights(interp_method, projected_grid, field_of_view): function apply_antialiasing_if_needed (line 301) | def apply_antialiasing_if_needed(interp_method, support_sz, scale_factor, function fw_ceil (line 315) | def fw_ceil(x, fw): function fw_cat (line 322) | def fw_cat(x, fw): function fw_swapaxes (line 329) | def fw_swapaxes(x, ax_1, ax_2, fw): function fw_set_device (line 335) | def fw_set_device(x, device, fw): FILE: code/synthetic/bsrt/scripts/cal_mean_std.py function main (line 9) | def main(): FILE: code/synthetic/bsrt/scripts/download_burstsr_dataset.py function download_burstsr_dataset (line 8) | def download_burstsr_dataset(download_path): function main (line 59) | def main(): FILE: code/synthetic/bsrt/scripts/evaluate_burstsr_val.py class SimpleBaseline (line 8) | class SimpleBaseline: method __init__ (line 9) | def __init__(self): method __call__ (line 12) | def __call__(self, burst): function main (line 19) | def main(): FILE: code/synthetic/bsrt/scripts/save_results_synburst_val.py class SimpleBaseline (line 9) | class SimpleBaseline: method __init__ (line 10) | def __init__(self): method __call__ (line 13) | def __call__(self, burst): function main (line 20) | def main(): FILE: code/synthetic/bsrt/scripts/test_burstsr_dataset.py function main (line 11) | def main(): FILE: code/synthetic/bsrt/scripts/test_synthetic_bursts.py function main (line 11) | def main(): FILE: code/synthetic/bsrt/test.py function ttaup (line 23) | def ttaup(burst): function ttadown (line 32) | def ttadown(bursts): function main (line 39) | def main(): function main_worker (line 43) | def main_worker(local_rank, nprocs, args): FILE: code/synthetic/bsrt/test_synburst.py function ttaup (line 29) | def ttaup(burst): function ttadown (line 35) | def ttadown(bursts): function main (line 42) | def main(): function main_worker (line 46) | def main_worker(local_rank, nprocs, args): FILE: code/synthetic/bsrt/trainer.py class Trainer (line 47) | class Trainer(): method __init__ (line 48) | def __init__(self, args, train_loader, train_sampler, valid_loader, my... method train (line 118) | def train(self): method test (line 233) | def test(self, print_time=False): method save_model (line 319) | def save_model(self, filename): method prepare (line 331) | def prepare(self, *args): method terminate (line 341) | def terminate(self): FILE: code/synthetic/bsrt/utility.py function reduce_mean (line 25) | def reduce_mean(tensor, nprocs): function gradient (line 31) | def gradient(data): function smooth_grad_1st (line 37) | def smooth_grad_1st(flo, image, alpha): function smooth_loss (line 48) | def smooth_loss(flow, img): function setup (line 53) | def setup(rank, world_size): function cleanup (line 80) | def cleanup(): function mkdir (line 84) | def mkdir(path): class timer (line 89) | class timer(): method __init__ (line 90) | def __init__(self): method tic (line 94) | def tic(self): method toc (line 97) | def toc(self, restart=False): method hold (line 102) | def hold(self): method release (line 105) | def release(self): method reset (line 111) | def reset(self): class checkpoint (line 115) | class checkpoint(): method __init__ (line 116) | def __init__(self, args): method get_path (line 153) | def get_path(self, *subdir): method save (line 156) | def save(self, trainer, epoch, is_best=False): method add_log (line 165) | def add_log(self, log): method write_log (line 168) | def write_log(self, log, refresh=False): method done (line 175) | def done(self): method plot_psnr (line 178) | def plot_psnr(self, epoch): method begin_background (line 197) | def begin_background(self): method end_background (line 214) | def end_background(self): method save_results (line 219) | def save_results(self, dataset, filename, save_list, scale): function quantize (line 233) | def quantize(img, rgb_range): function calc_psnr (line 238) | def calc_psnr(sr, hr, scale, rgb_range, dataset=None): function make_optimizer (line 257) | def make_optimizer(args, target): function write_gray_to_tfboard (line 313) | def write_gray_to_tfboard(img): function bayer_unify (line 333) | def bayer_unify(raw, input_pattern, target_pattern, mode) -> np.ndarray: function bayer_aug (line 365) | def bayer_aug(raw, flip_h=False, flip_w=False, transpose=False, input_pa... FILE: code/synthetic/bsrt/utils/data_format_utils.py function numpy_to_torch (line 6) | def numpy_to_torch(a: np.ndarray): function torch_to_numpy (line 10) | def torch_to_numpy(a: torch.Tensor): function torch_to_npimage (line 14) | def torch_to_npimage(a: torch.Tensor, unnormalize=True): function npimage_to_torch (line 23) | def npimage_to_torch(a, normalize=True, input_bgr=True): function convert_dict (line 34) | def convert_dict(base_dict, batch_sz): FILE: code/synthetic/bsrt/utils/debayer.py class Debayer3x3 (line 5) | class Debayer3x3(torch.nn.Module): method __init__ (line 29) | def __init__(self): method forward (line 71) | def forward(self, x): class Debayer2x2 (line 91) | class Debayer2x2(torch.nn.Module): method __init__ (line 99) | def __init__(self): method forward (line 115) | def forward(self, x): class DebayerSplit (line 133) | class DebayerSplit(torch.nn.Module): method __init__ (line 140) | def __init__(self): method forward (line 151) | def forward(self, x): FILE: code/synthetic/bsrt/utils/interp_methods.py function set_framework_dependencies (line 17) | def set_framework_dependencies(x): function support_sz (line 28) | def support_sz(sz): function cubic (line 35) | def cubic(x): function lanczos2 (line 45) | def lanczos2(x): function lanczos3 (line 51) | def lanczos3(x): function linear (line 57) | def linear(x): function box (line 63) | def box(x): FILE: code/synthetic/bsrt/utils/metrics.py class MSSSIMLoss (line 15) | class MSSSIMLoss(nn.Module): method __init__ (line 16) | def __init__(self, boundary_ignore=None): method forward (line 21) | def forward(self, pred, gt, valid=None): class CharbonnierLoss (line 33) | class CharbonnierLoss(nn.Module): method __init__ (line 34) | def __init__(self, boundary_ignore=None): method forward (line 39) | def forward(self, pred, gt, valid=None): class L1 (line 51) | class L1(nn.Module): method __init__ (line 52) | def __init__(self, boundary_ignore=None): method forward (line 56) | def forward(self, pred, gt, valid=None): class L2 (line 78) | class L2(nn.Module): method __init__ (line 79) | def __init__(self, boundary_ignore=None): method forward (line 84) | def forward(self, pred, gt, valid=None): class PSNR (line 110) | class PSNR(nn.Module): method __init__ (line 111) | def __init__(self, boundary_ignore=None, max_value=1.0): method psnr (line 116) | def psnr(self, pred, gt, valid=None): method forward (line 123) | def forward(self, pred, gt, valid=None): class AlignedL1 (line 136) | class AlignedL1(nn.Module): method __init__ (line 137) | def __init__(self, alignment_net, sr_factor=4, boundary_ignore=None): method forward (line 145) | def forward(self, pred, gt, burst_input): class AlignedL2 (line 186) | class AlignedL2(nn.Module): method __init__ (line 187) | def __init__(self, alignment_net, sr_factor=4, boundary_ignore=None): method forward (line 196) | def forward(self, pred, gt, burst_input): class AlignedPSNR (line 243) | class AlignedPSNR(nn.Module): method __init__ (line 244) | def __init__(self, alignment_net, sr_factor=4, boundary_ignore=None, m... method psnr (line 249) | def psnr(self, pred, gt, burst_input): method forward (line 256) | def forward(self, pred, gt, burst_input): class AlignedSSIM (line 265) | class AlignedSSIM(nn.Module): method __init__ (line 266) | def __init__(self, alignment_net, sr_factor=4, boundary_ignore=None): method _ssim (line 274) | def _ssim(self, pred, gt, burst_input): method forward (line 313) | def forward(self, pred, gt, burst_input): class AlignedLPIPS (line 319) | class AlignedLPIPS(nn.Module): method __init__ (line 320) | def __init__(self, alignment_net, sr_factor=4, boundary_ignore=None): method _lpips (line 329) | def _lpips(self, pred, gt, burst_input): method forward (line 363) | def forward(self, pred, gt, burst_input): FILE: code/synthetic/bsrt/utils/postprocessing_functions.py class SimplePostProcess (line 7) | class SimplePostProcess: method __init__ (line 8) | def __init__(self, gains=True, ccm=True, gamma=True, smoothstep=True, ... method process (line 15) | def process(self, image, meta_info): function process_linear_image_rgb (line 20) | def process_linear_image_rgb(image, meta_info, gains=True, ccm=True, gam... class BurstSRPostProcess (line 40) | class BurstSRPostProcess: method __init__ (line 41) | def __init__(self, no_white_balance=False, gamma=True, smoothstep=True... method process (line 47) | def process(self, image, meta_info, external_norm_factor=None): function process_burstsr_image_rgb (line 53) | def process_burstsr_image_rgb(im, meta_info, return_np=False, external_n... FILE: code/synthetic/bsrt/utils/resize_right.py class NoneClass (line 6) | class NoneClass: function resize (line 29) | def resize(input, scale_factors=None, out_shape=None, class ResizeLayer (line 80) | class ResizeLayer(nnModuleWrapped): method __init__ (line 81) | def __init__(self, in_shape, scale_factors=None, out_shape=None, method forward (line 132) | def forward(self, input): function prepare_weights_and_field_of_view_1d (line 147) | def prepare_weights_and_field_of_view_1d(dim, scale_factor, in_sz, out_sz, function apply_weights (line 174) | def apply_weights(input, field_of_view, weights, dim, n_dims, fw): function set_scale_and_out_sz (line 211) | def set_scale_and_out_sz(in_shape, out_shape, scale_factors, fw): function get_projected_grid (line 251) | def get_projected_grid(in_sz, out_sz, scale_factor, fw, device=None): function get_field_of_view (line 266) | def get_field_of_view(projected_grid, cur_support_sz, in_sz, fw, eps): function get_weights (line 288) | def get_weights(interp_method, projected_grid, field_of_view): function apply_antialiasing_if_needed (line 301) | def apply_antialiasing_if_needed(interp_method, support_sz, scale_factor, function fw_ceil (line 315) | def fw_ceil(x, fw): function fw_cat (line 322) | def fw_cat(x, fw): function fw_swapaxes (line 329) | def fw_swapaxes(x, ax_1, ax_2, fw): function fw_set_device (line 335) | def fw_set_device(x, device, fw): FILE: code/synthetic/bsrt/utils/spatial_color_alignment.py function gauss_1d (line 6) | def gauss_1d(sz, sigma, center, end_pad=0, density=False): function gauss_2d (line 15) | def gauss_2d(sz, sigma, center, end_pad=(0, 0), density=False): function get_gaussian_kernel (line 29) | def get_gaussian_kernel(sd): function apply_kernel (line 38) | def apply_kernel(im, ksz, gauss_kernel): function match_colors (line 48) | def match_colors(im_ref, im_q, im_test, ksz, gauss_kernel): FILE: code/synthetic/bsrt/utils/stn.py class SpatialTransformer (line 6) | class SpatialTransformer(nn.Module): method __init__ (line 12) | def __init__(self, size, mode='bilinear'): method forward (line 31) | def forward(self, src, flow): FILE: code/synthetic/bsrt/utils/warp.py function warp (line 6) | def warp(feat, flow, mode='bilinear', padding_mode='zeros'):