SYMBOL INDEX (150 symbols across 13 files) FILE: calculate_PSNR_SSIM.py function main (line 12) | def main(): function calculate_psnr (line 75) | def calculate_psnr(img1, img2): function ssim (line 85) | def ssim(img1, img2): function calculate_ssim (line 108) | def calculate_ssim(img1, img2): function bgr2ycbcr (line 129) | def bgr2ycbcr(img, only_y=True): FILE: datasets.py function to_rgb (line 9) | def to_rgb(image): class Hinet_Dataset (line 15) | class Hinet_Dataset(Dataset): method __init__ (line 16) | def __init__(self, transforms_=None, mode="train"): method __getitem__ (line 27) | def __getitem__(self, index): method __len__ (line 37) | def __len__(self): FILE: hinet.py class Hinet (line 5) | class Hinet(nn.Module): method __init__ (line 7) | def __init__(self): method forward (line 28) | def forward(self, x, rev=False): FILE: invblock.py class INV_block (line 8) | class INV_block(nn.Module): method __init__ (line 9) | def __init__(self, subnet_constructor=ResidualDenseBlock_out, clamp=c.... method e (line 22) | def e(self, s): method forward (line 25) | def forward(self, x, rev=False): FILE: model.py class Model (line 7) | class Model(nn.Module): method __init__ (line 8) | def __init__(self): method forward (line 13) | def forward(self, x, rev=False): function init_model (line 24) | def init_model(mod): FILE: modules/Unet_common.py function default_conv (line 9) | def default_conv(in_channels, out_channels, kernel_size, bias=True, dila... function default_conv1 (line 20) | def default_conv1(in_channels, out_channels, kernel_size, bias=True, gro... function default_conv3d (line 30) | def default_conv3d(in_channels, out_channels, kernel_size, t_kernel=3, b... function channel_shuffle (line 42) | def channel_shuffle(x, groups): function pixel_down_shuffle (line 58) | def pixel_down_shuffle(x, downsacale_factor): function sp_init (line 73) | def sp_init(x): function dwt_init3d (line 85) | def dwt_init3d(x): function dwt_init (line 100) | def dwt_init(x): function iwt_init (line 115) | def iwt_init(x): class ResidualDenseBlock (line 136) | class ResidualDenseBlock(nn.Module): method __init__ (line 137) | def __init__(self, nf=64, gc=32, kernel_size = 3, bias=True, use_snorm... method forward (line 158) | def forward(self, x): class RRDB (line 167) | class RRDB(nn.Module): method __init__ (line 170) | def __init__(self, nf, gc=32, use_snorm=False): method forward (line 176) | def forward(self, x): class RRDBblock (line 182) | class RRDBblock(nn.Module): method __init__ (line 185) | def __init__(self, nf, gc=32, nb=23, use_snorm=False): method forward (line 195) | def forward(self, x): class Channel_Shuffle (line 199) | class Channel_Shuffle(nn.Module): method __init__ (line 200) | def __init__(self, conv_groups): method forward (line 205) | def forward(self, x): class SP (line 208) | class SP(nn.Module): method __init__ (line 209) | def __init__(self): method forward (line 213) | def forward(self, x): class Pixel_Down_Shuffle (line 216) | class Pixel_Down_Shuffle(nn.Module): method __init__ (line 217) | def __init__(self): method forward (line 221) | def forward(self, x): class DWT (line 224) | class DWT(nn.Module): method __init__ (line 225) | def __init__(self): method forward (line 229) | def forward(self, x): class DWT3d (line 232) | class DWT3d(nn.Module): method __init__ (line 233) | def __init__(self): method forward (line 237) | def forward(self, x): class IWT (line 240) | class IWT(nn.Module): method __init__ (line 241) | def __init__(self): method forward (line 245) | def forward(self, x): class MeanShift (line 249) | class MeanShift(nn.Conv2d): method __init__ (line 250) | def __init__(self, rgb_range, rgb_mean, rgb_std, sign=-1): class MeanShift2 (line 261) | class MeanShift2(nn.Conv2d): method __init__ (line 262) | def __init__(self, rgb_range, rgb_mean, rgb_std, sign=-1): class BasicBlock (line 273) | class BasicBlock(nn.Sequential): method __init__ (line 274) | def __init__( class Block3d (line 292) | class Block3d(nn.Sequential): method __init__ (line 293) | def __init__( method forward (line 309) | def forward(self, x): class BBlock (line 313) | class BBlock(nn.Module): method __init__ (line 314) | def __init__( method forward (line 329) | def forward(self, x): class DBlock_com (line 333) | class DBlock_com(nn.Module): method __init__ (line 334) | def __init__( method forward (line 351) | def forward(self, x): class DBlock_inv (line 355) | class DBlock_inv(nn.Module): method __init__ (line 356) | def __init__( method forward (line 374) | def forward(self, x): class DBlock_com1 (line 378) | class DBlock_com1(nn.Module): method __init__ (line 379) | def __init__( method forward (line 397) | def forward(self, x): class DBlock_inv1 (line 401) | class DBlock_inv1(nn.Module): method __init__ (line 402) | def __init__( method forward (line 420) | def forward(self, x): class DBlock_com2 (line 424) | class DBlock_com2(nn.Module): method __init__ (line 425) | def __init__( method forward (line 443) | def forward(self, x): class DBlock_inv2 (line 447) | class DBlock_inv2(nn.Module): method __init__ (line 448) | def __init__( method forward (line 466) | def forward(self, x): class ShuffleBlock (line 470) | class ShuffleBlock(nn.Module): method __init__ (line 471) | def __init__( method forward (line 486) | def forward(self, x): class DWBlock (line 491) | class DWBlock(nn.Module): method __init__ (line 492) | def __init__( method forward (line 510) | def forward(self, x): class ResBlock (line 514) | class ResBlock(nn.Module): method __init__ (line 515) | def __init__( method forward (line 529) | def forward(self, x): class Block (line 535) | class Block(nn.Module): method __init__ (line 536) | def __init__( method forward (line 550) | def forward(self, x): class Upsampler (line 556) | class Upsampler(nn.Sequential): method __init__ (line 557) | def __init__(self, conv, scale, n_feat, bn=False, act=False, bias=True... class VGG_conv0 (line 576) | class VGG_conv0(nn.Module): method __init__ (line 577) | def __init__(self, in_nc, nf): method forward (line 608) | def forward(self, x): class VGG_conv1 (line 627) | class VGG_conv1(nn.Module): method __init__ (line 628) | def __init__(self, in_nc, nf): method forward (line 659) | def forward(self, x): class VGG_conv2 (line 677) | class VGG_conv2(nn.Module): method __init__ (line 678) | def __init__(self, in_nc, nf): method forward (line 709) | def forward(self, x): FILE: modules/module_util.py function initialize_weights (line 7) | def initialize_weights(net_l, scale=1): function make_layer (line 27) | def make_layer(block, n_layers): class ResidualBlock_noBN (line 34) | class ResidualBlock_noBN(nn.Module): method __init__ (line 40) | def __init__(self, nf=64): method forward (line 48) | def forward(self, x): function flow_warp (line 55) | def flow_warp(x, flow, interp_mode='bilinear', padding_mode='zeros'): FILE: rrdb_denselayer.py class ResidualDenseBlock_out (line 7) | class ResidualDenseBlock_out(nn.Module): method __init__ (line 8) | def __init__(self, input, output, bias=True): method forward (line 19) | def forward(self, x): FILE: test.py function load (line 16) | def load(name): function gauss_noise (line 26) | def gauss_noise(shape): function computePSNR (line 35) | def computePSNR(origin,pred): FILE: train.py function gauss_noise (line 19) | def gauss_noise(shape): function guide_loss (line 27) | def guide_loss(output, bicubic_image): function reconstruction_loss (line 33) | def reconstruction_loss(rev_input, input): function low_frequency_loss (line 39) | def low_frequency_loss(ll_input, gt_input): function get_parameter_number (line 46) | def get_parameter_number(net): function computePSNR (line 52) | def computePSNR(origin,pred): function load (line 63) | def load(name): FILE: train_logging.py function gauss_noise (line 20) | def gauss_noise(shape): function guide_loss (line 28) | def guide_loss(output, bicubic_image): function reconstruction_loss (line 34) | def reconstruction_loss(rev_input, input): function low_frequency_loss (line 40) | def low_frequency_loss(ll_input, gt_input): function get_parameter_number (line 47) | def get_parameter_number(net): function computePSNR (line 53) | def computePSNR(origin,pred): function load (line 64) | def load(name): FILE: util.py function get_timestamp (line 5) | def get_timestamp(): function setup_logger (line 8) | def setup_logger(logger_name, root, phase, level=logging.INFO, screen=Fa... FILE: viz.py class Visualizer (line 14) | class Visualizer: method __init__ (line 15) | def __init__(self, loss_labels): method update_losses (line 38) | def update_losses(self, losses, *args): method update_images (line 47) | def update_images(self, *img_list): method update_hist (line 70) | def update_hist(self, *args): method update_running (line 73) | def update_running(self, *args): function show_loss (line 79) | def show_loss(losses, logscale=False): function show_imgs (line 82) | def show_imgs(*imgs): function show_hist (line 85) | def show_hist(data): function signal_start (line 88) | def signal_start(): function signal_stop (line 91) | def signal_stop(): function close (line 94) | def close():