SYMBOL INDEX (447 symbols across 65 files) FILE: KDSR-GAN/Metric/DISTS/DISTS_pytorch/DISTS_pt.py class L2pooling (line 11) | class L2pooling(nn.Module): method __init__ (line 12) | def __init__(self, filter_size=5, stride=2, channels=None, pad_off=0): method forward (line 22) | def forward(self, input): class DISTS (line 27) | class DISTS(torch.nn.Module): method __init__ (line 28) | def __init__(self, load_weights=True): method forward_once (line 68) | def forward_once(self, x): method forward (line 82) | def forward(self, x, y, require_grad=False, batch_average=False): function prepare_image (line 115) | def prepare_image(image, resize=True): FILE: KDSR-GAN/Metric/DISTS/DISTS_tensorflow/DISTS_tf.py class DISTS (line 13) | class DISTS(): method __init__ (line 14) | def __init__(self): method get_features (line 25) | def get_features(self, img): method conv_layer (line 53) | def conv_layer(self, input, name): method pool_layer (line 61) | def pool_layer(self, input, name): method get_conv_filter (line 68) | def get_conv_filter(self, name): method get_bias (line 71) | def get_bias(self, name): method get_score (line 74) | def get_score(self, img1, img2): FILE: KDSR-GAN/Metric/LPIPS.py function main (line 11) | def main(): FILE: KDSR-GAN/Metric/PSNR.py function main (line 12) | def main(): FILE: KDSR-GAN/Metric/dists.py class L2pooling (line 13) | class L2pooling(nn.Module): method __init__ (line 14) | def __init__(self, filter_size=5, stride=2, channels=None, pad_off=0): method forward (line 24) | def forward(self, input): class DISTS (line 29) | class DISTS(torch.nn.Module): method __init__ (line 30) | def __init__(self, load_weights=True): method forward_once (line 70) | def forward_once(self, x): method forward (line 84) | def forward(self, x, y, require_grad=False, batch_average=False): function prepare_image (line 117) | def prepare_image(image, resize=True): FILE: KDSR-GAN/kdsrgan/archs/ST_arch.py class IDR_DDC (line 8) | class IDR_DDC(nn.Module): method __init__ (line 9) | def __init__(self, channels_in, channels_out, kernel_size, reduction): method forward (line 21) | def forward(self, x): class IDR_DCRB (line 38) | class IDR_DCRB(nn.Module): method __init__ (line 39) | def __init__(self, conv, n_feat, kernel_size, reduction): method forward (line 46) | def forward(self, x): class DAG (line 59) | class DAG(nn.Module): method __init__ (line 60) | def __init__(self, conv, n_feat, kernel_size, reduction, n_blocks): method forward (line 71) | def forward(self, x): class KDSR (line 83) | class KDSR(nn.Module): method __init__ (line 84) | def __init__(self, n_feats=128, scale=4,n_sr_blocks = 42, rgb_range=1,... method forward (line 111) | def forward(self, x, k_v): class KD_IDE (line 135) | class KD_IDE(nn.Module): method __init__ (line 136) | def __init__(self,n_feats = 128, n_encoder_res = 6): method forward (line 169) | def forward(self, x): class BlindSR_ST (line 178) | class BlindSR_ST(nn.Module): method __init__ (line 179) | def __init__(self, n_feats=128, n_encoder_res=6, scale=4,n_sr_blocks=4... method forward (line 190) | def forward(self, x): FILE: KDSR-GAN/kdsrgan/archs/TA_arch.py class IDR_DDC (line 7) | class IDR_DDC(nn.Module): method __init__ (line 8) | def __init__(self, channels_in, channels_out, kernel_size, reduction): method forward (line 20) | def forward(self, x): class IDR_DCRB (line 37) | class IDR_DCRB(nn.Module): method __init__ (line 38) | def __init__(self, conv, n_feat, kernel_size, reduction): method forward (line 45) | def forward(self, x): class DAG (line 58) | class DAG(nn.Module): method __init__ (line 59) | def __init__(self, conv, n_feat, kernel_size, reduction, n_blocks): method forward (line 70) | def forward(self, x): class KDSR (line 84) | class KDSR(nn.Module): method __init__ (line 85) | def __init__(self, n_feats=128, scale=4,n_sr_blocks = 42, rgb_range=1,... method forward (line 112) | def forward(self, x, k_v): class KD_IDE (line 136) | class KD_IDE(nn.Module): method __init__ (line 137) | def __init__(self,n_feats = 128, n_encoder_res = 6): method forward (line 170) | def forward(self, x): class BlindSR_TA (line 179) | class BlindSR_TA(nn.Module): method __init__ (line 180) | def __init__(self, n_feats=128, n_encoder_res=6, scale=4,n_sr_blocks=4... method forward (line 191) | def forward(self, x, gt): FILE: KDSR-GAN/kdsrgan/archs/common.py function default_conv (line 7) | def default_conv(in_channels, out_channels, kernel_size, bias=True): class ResBlock (line 10) | class ResBlock(nn.Module): method __init__ (line 11) | def __init__( method forward (line 27) | def forward(self, x): class MeanShift (line 33) | class MeanShift(nn.Conv2d): method __init__ (line 34) | def __init__(self, rgb_range, rgb_mean, rgb_std, sign=-1): class Upsampler (line 45) | class Upsampler(nn.Sequential): method __init__ (line 46) | def __init__(self, conv, scale, n_feat, act=False, bias=True): FILE: KDSR-GAN/kdsrgan/archs/discriminator_arch.py class UNetDiscriminatorSN (line 8) | class UNetDiscriminatorSN(nn.Module): method __init__ (line 19) | def __init__(self, num_in_ch, num_feat=64, skip_connection=True): method forward (line 38) | def forward(self, x): FILE: KDSR-GAN/kdsrgan/archs/srvgg_arch.py class SRVGGNetCompact (line 7) | class SRVGGNetCompact(nn.Module): method __init__ (line 22) | def __init__(self, num_in_ch=3, num_out_ch=3, num_feat=64, num_conv=16... method forward (line 60) | def forward(self, x): FILE: KDSR-GAN/kdsrgan/data/kdsrgan_dataset.py class KDSRGANDataset (line 17) | class KDSRGANDataset(data.Dataset): method __init__ (line 35) | def __init__(self, opt): method __getitem__ (line 83) | def __getitem__(self, index): method __len__ (line 191) | def __len__(self): FILE: KDSR-GAN/kdsrgan/data/kdsrgan_paired_dataset.py class KDSRGANPairedDataset (line 11) | class KDSRGANPairedDataset(data.Dataset): method __init__ (line 41) | def __init__(self, opt): method __getitem__ (line 75) | def __getitem__(self, index): method __len__ (line 107) | def __len__(self): FILE: KDSR-GAN/kdsrgan/losses/my_loss.py class KDLoss (line 8) | class KDLoss(nn.Module): method __init__ (line 14) | def __init__(self, loss_weight=1.0, temperature = 0.15): method forward (line 20) | def forward(self, T_fea, S_fea): FILE: KDSR-GAN/kdsrgan/models/kdsrgan_ST_model.py class KDSRGANSTModel (line 18) | class KDSRGANSTModel(SRGANModel): method __init__ (line 26) | def __init__(self, opt): method init_training_settings (line 58) | def init_training_settings(self): method _dequeue_and_enqueue (line 69) | def _dequeue_and_enqueue(self): method feed_data (line 106) | def feed_data(self, data): method nondist_validation (line 224) | def nondist_validation(self, dataloader, current_iter, tb_logger, save... method test (line 230) | def test(self): method optimize_parameters (line 241) | def optimize_parameters(self, current_iter): FILE: KDSR-GAN/kdsrgan/models/kdsrgan_TA_model.py class KDSRGANTAModel (line 15) | class KDSRGANTAModel(SRGANModel): method __init__ (line 23) | def __init__(self, opt): method _dequeue_and_enqueue (line 30) | def _dequeue_and_enqueue(self): method feed_data (line 67) | def feed_data(self, data): method nondist_validation (line 185) | def nondist_validation(self, dataloader, current_iter, tb_logger, save... method optimize_parameters (line 191) | def optimize_parameters(self, current_iter): FILE: KDSR-GAN/kdsrgan/models/kdsrnet_ST_model.py class KDSRNetSTModel (line 19) | class KDSRNetSTModel(SRModel): method __init__ (line 27) | def __init__(self, opt): method setup_optimizers (line 59) | def setup_optimizers(self): method setup_schedulers (line 73) | def setup_schedulers(self): method init_training_settings (line 87) | def init_training_settings(self): method _dequeue_and_enqueue (line 131) | def _dequeue_and_enqueue(self): method feed_data (line 168) | def feed_data(self, data): method nondist_validation (line 285) | def nondist_validation(self, dataloader, current_iter, tb_logger, save... method test (line 291) | def test(self): method optimize_parameters (line 303) | def optimize_parameters(self, current_iter): FILE: KDSR-GAN/kdsrgan/models/kdsrnet_TA_model.py class KDSRNetTAModel (line 15) | class KDSRNetTAModel(SRModel): method __init__ (line 23) | def __init__(self, opt): method _dequeue_and_enqueue (line 32) | def _dequeue_and_enqueue(self): method feed_data (line 69) | def feed_data(self, data): method nondist_validation (line 187) | def nondist_validation(self, dataloader, current_iter, tb_logger, save... method test (line 193) | def test(self): method optimize_parameters (line 205) | def optimize_parameters(self, current_iter): FILE: KDSR-GAN/kdsrgan/utils.py class RealESRGANer (line 14) | class RealESRGANer(): method __init__ (line 29) | def __init__(self, method pre_process (line 68) | def pre_process(self, img): method process (line 93) | def process(self): method tile_process (line 97) | def tile_process(self): method post_process (line 162) | def post_process(self): method enhance (line 174) | def enhance(self, img, outscale=None, alpha_upsampler='realesrgan'): class PrefetchReader (line 246) | class PrefetchReader(threading.Thread): method __init__ (line 254) | def __init__(self, img_list, num_prefetch_queue): method run (line 259) | def run(self): method __next__ (line 266) | def __next__(self): method __iter__ (line 272) | def __iter__(self): class IOConsumer (line 276) | class IOConsumer(threading.Thread): method __init__ (line 278) | def __init__(self, opt, que, qid): method run (line 284) | def run(self): FILE: KDSR-GAN/scripts/extract_subimages.py function main (line 12) | def main(args): function extract_subimages (line 43) | def extract_subimages(opt): function worker (line 74) | def worker(path, opt): FILE: KDSR-GAN/scripts/extract_subimages_DF2K.py function main (line 12) | def main(args): function extract_subimages (line 43) | def extract_subimages(opt): function worker (line 74) | def worker(path, opt): FILE: KDSR-GAN/scripts/generate_meta_info.py function main (line 7) | def main(args): FILE: KDSR-GAN/scripts/generate_meta_info_DF2K.py function main (line 7) | def main(args): FILE: KDSR-GAN/scripts/generate_meta_info_OST.py function main (line 7) | def main(args): FILE: KDSR-GAN/scripts/generate_meta_info_pairdata.py function main (line 6) | def main(args): FILE: KDSR-GAN/scripts/generate_multiscale_DF2K.py function main (line 7) | def main(args): FILE: KDSR-GAN/scripts/pytorch2onnx.py function main (line 7) | def main(args): FILE: KDSR-GAN/setup.py function readme (line 12) | def readme(): function get_git_hash (line 18) | def get_git_hash(): function get_hash (line 43) | def get_hash(): function write_version_py (line 52) | def write_version_py(): function get_version (line 69) | def get_version(): function get_requirements (line 75) | def get_requirements(filename='requirements.txt'): FILE: KDSR-GAN/tests/test_dataset.py function test_realesrgan_dataset (line 8) | def test_realesrgan_dataset(): function test_realesrgan_paired_dataset (line 82) | def test_realesrgan_paired_dataset(): FILE: KDSR-GAN/tests/test_discriminator_arch.py function test_unetdiscriminatorsn (line 6) | def test_unetdiscriminatorsn(): FILE: KDSR-GAN/tests/test_model.py function test_realesrnet_model (line 12) | def test_realesrnet_model(): function test_realesrgan_model (line 64) | def test_realesrgan_model(): FILE: KDSR-GAN/tests/test_utils.py function test_realesrganer (line 7) | def test_realesrganer(): FILE: KDSR-classic/cal_parms.py function count_param (line 1) | def count_param(model): FILE: KDSR-classic/data/__init__.py class MyConcatDataset (line 7) | class MyConcatDataset(ConcatDataset): method __init__ (line 8) | def __init__(self, datasets): method set_scale (line 12) | def set_scale(self, idx_scale): class Data (line 16) | class Data: method __init__ (line 17) | def __init__(self, args): FILE: KDSR-classic/data/benchmark.py class Benchmark (line 6) | class Benchmark(srdata.SRData): method __init__ (line 7) | def __init__(self, args, name='', train=True): method _set_filesystem (line 12) | def _set_filesystem(self, dir_data): FILE: KDSR-classic/data/common.py function get_patch (line 9) | def get_patch(hr, patch_size=48, scale=2, multi=False, input_large=False): function set_channel (line 22) | def set_channel(hr, n_channels=3): function np2Tensor (line 38) | def np2Tensor(hr, rgb_range=255): function augment (line 49) | def augment(hr, hflip=True, rot=True): FILE: KDSR-classic/data/common2.py function get_patch (line 9) | def get_patch(*args, patch_size=96, scale=2, multi=False, input_large=Fa... function set_channel (line 36) | def set_channel(*args, n_channels=3): function np2Tensor (line 52) | def np2Tensor(*args, rgb_range=255): function augment (line 63) | def augment(*args, hflip=True, rot=True): FILE: KDSR-classic/data/df2k.py class DF2K (line 5) | class DF2K(multiscalesrdata.SRData): method __init__ (line 6) | def __init__(self, args, name='DF2K', train=True, benchmark=False): method _scan (line 19) | def _scan(self): method _set_filesystem (line 25) | def _set_filesystem(self, dir_data): FILE: KDSR-classic/data/multiscalesrdata.py class SRData (line 14) | class SRData(data.Dataset): method __init__ (line 15) | def __init__(self, args, name='', train=True, benchmark=False): method _scan (line 69) | def _scan(self): method _set_filesystem (line 85) | def _set_filesystem(self, dir_data): method _check_and_load (line 92) | def _check_and_load(self, ext, img, f, verbose=True): method __getitem__ (line 99) | def __getitem__(self, idx): method __len__ (line 107) | def __len__(self): method _get_index (line 113) | def _get_index(self, idx): method _load_file (line 119) | def _load_file(self, idx): method get_patch (line 132) | def get_patch(self, hr): method set_scale (line 151) | def set_scale(self, idx_scale): FILE: KDSR-classic/data/multiscalesrdata2.py class SRData (line 14) | class SRData(data.Dataset): method __init__ (line 15) | def __init__(self, args, name='', train=True, benchmark=False): method _scan (line 69) | def _scan(self): method _set_filesystem (line 86) | def _set_filesystem(self, dir_data): method _check_and_load (line 93) | def _check_and_load(self, ext, img, f, verbose=True): method __getitem__ (line 100) | def __getitem__(self, idx): method __len__ (line 108) | def __len__(self): method _get_index (line 114) | def _get_index(self, idx): method _load_file (line 120) | def _load_file(self, idx): method get_patch (line 137) | def get_patch(self, lr, hr): method set_scale (line 154) | def set_scale(self, idx_scale): FILE: KDSR-classic/dataloader.py function _ms_loop (line 20) | def _ms_loop(dataset, index_queue, data_queue, collate_fn, scale, seed, ... class _MSDataLoaderIter (line 46) | class _MSDataLoaderIter(_DataLoaderIter): method __init__ (line 47) | def __init__(self, loader): class MSDataLoader (line 117) | class MSDataLoader(DataLoader): method __init__ (line 118) | def __init__( method __iter__ (line 133) | def __iter__(self): FILE: KDSR-classic/loss/__init__.py class Loss (line 13) | class Loss(nn.modules.loss._Loss): method __init__ (line 14) | def __init__(self, args, ckp): method forward (line 70) | def forward(self, sr, hr): method step (line 87) | def step(self): method start_log (line 92) | def start_log(self): method end_log (line 95) | def end_log(self, n_batches): method display_loss (line 98) | def display_loss(self, batch): method plot_loss (line 106) | def plot_loss(self, apath, epoch): method get_loss_module (line 120) | def get_loss_module(self): method save (line 126) | def save(self, apath): method load (line 130) | def load(self, apath, cpu=False): FILE: KDSR-classic/loss/adversarial.py class Adversarial (line 11) | class Adversarial(nn.Module): method __init__ (line 12) | def __init__(self, args, gan_type): method forward (line 26) | def forward(self, fake, real): method state_dict (line 78) | def state_dict(self, *args, **kwargs): FILE: KDSR-classic/loss/discriminator.py class Discriminator (line 5) | class Discriminator(nn.Module): method __init__ (line 6) | def __init__(self, args, gan_type='GAN'): method forward (line 40) | def forward(self, x): FILE: KDSR-classic/loss/vgg.py class VGG (line 9) | class VGG(nn.Module): method __init__ (line 10) | def __init__(self, conv_index, rgb_range=1): method forward (line 24) | def forward(self, sr, hr): FILE: KDSR-classic/main_anisonoise_stage3.py function count_param (line 10) | def count_param(model): FILE: KDSR-classic/main_iso_stage3.py function count_param (line 10) | def count_param(model): FILE: KDSR-classic/model/__init__.py class Model (line 8) | class Model(nn.Module): method __init__ (line 9) | def __init__(self, args, ckp): method forward (line 38) | def forward(self, lr,lr_bic): method get_model (line 51) | def get_model(self): method state_dict (line 57) | def state_dict(self, **kwargs): method save (line 61) | def save(self, apath, epoch, is_best=False): method load (line 79) | def load(self, apath, pre_train='.', resume=-1, cpu=False): method forward_chop (line 104) | def forward_chop(self, x, shave=10, min_size=160000): method forward_x8 (line 145) | def forward_x8(self, x, forward_function): FILE: KDSR-classic/model/blindsr.py function make_model (line 8) | def make_model(args): class CZSR (line 12) | class CZSR(nn.Module): method __init__ (line 13) | def __init__(self, args, conv=common.default_conv): method forward (line 60) | def forward(self, lr,lr_bic): FILE: KDSR-classic/model/common.py function default_conv (line 7) | def default_conv(in_channels, out_channels, kernel_size, bias=True): class ResBlock (line 10) | class ResBlock(nn.Module): method __init__ (line 11) | def __init__( method forward (line 27) | def forward(self, x): class MeanShift (line 33) | class MeanShift(nn.Conv2d): method __init__ (line 34) | def __init__(self, rgb_range, rgb_mean, rgb_std, sign=-1): class Upsampler (line 45) | class Upsampler(nn.Sequential): method __init__ (line 46) | def __init__(self, conv, scale, n_feat, act=False, bias=True): FILE: KDSR-classic/model_ST/__init__.py class Model (line 8) | class Model(nn.Module): method __init__ (line 9) | def __init__(self, args, ckp): method forward (line 38) | def forward(self, x): method get_model (line 51) | def get_model(self): method state_dict (line 57) | def state_dict(self, **kwargs): method save (line 61) | def save(self, apath, epoch, is_best=False): method load (line 79) | def load(self, apath, pre_train='.', resume=-1, cpu=False): method forward_chop (line 104) | def forward_chop(self, x, shave=10, min_size=160000): method forward_x8 (line 145) | def forward_x8(self, x, forward_function): FILE: KDSR-classic/model_ST/blindsr.py function make_model (line 7) | def make_model(args): class IDR_DDC (line 10) | class IDR_DDC(nn.Module): method __init__ (line 11) | def __init__(self, channels_in, channels_out, kernel_size, reduction): method forward (line 23) | def forward(self, x): class IDR_DCRB (line 40) | class IDR_DCRB(nn.Module): method __init__ (line 41) | def __init__(self, conv, n_feat, kernel_size, reduction): method forward (line 48) | def forward(self, x): class DAG (line 61) | class DAG(nn.Module): method __init__ (line 62) | def __init__(self, conv, n_feat, kernel_size, reduction, n_blocks): method forward (line 73) | def forward(self, x): class KDSR (line 85) | class KDSR(nn.Module): method __init__ (line 86) | def __init__(self, args, conv=common.default_conv): method forward (line 118) | def forward(self, x, k_v): class KD_IDE (line 142) | class KD_IDE(nn.Module): method __init__ (line 143) | def __init__(self,args): method forward (line 180) | def forward(self, x): class BlindSR (line 193) | class BlindSR(nn.Module): method __init__ (line 194) | def __init__(self, args): method forward (line 203) | def forward(self, x): FILE: KDSR-classic/model_ST/common.py function default_conv (line 7) | def default_conv(in_channels, out_channels, kernel_size, bias=True): class MeanShift (line 11) | class MeanShift(nn.Conv2d): method __init__ (line 12) | def __init__(self, rgb_range, rgb_mean, rgb_std, sign=-1): class Upsampler (line 23) | class Upsampler(nn.Sequential): method __init__ (line 24) | def __init__(self, conv, scale, n_feat, act=False, bias=True): FILE: KDSR-classic/model_TA/__init__.py class Model (line 8) | class Model(nn.Module): method __init__ (line 9) | def __init__(self, args, ckp): method forward (line 38) | def forward(self, x, deg_repre): method get_model (line 51) | def get_model(self): method state_dict (line 57) | def state_dict(self, **kwargs): method save (line 61) | def save(self, apath, epoch, is_best=False): method load (line 79) | def load(self, apath, pre_train='.', resume=-1, cpu=False): method forward_chop (line 104) | def forward_chop(self, x, deg_repre, shave=10, min_size=160000): method forward_x8 (line 156) | def forward_x8(self, x, forward_function): FILE: KDSR-classic/model_TA/blindsr.py function make_model (line 7) | def make_model(args): class IDR_DDC (line 10) | class IDR_DDC(nn.Module): method __init__ (line 11) | def __init__(self, channels_in, channels_out, kernel_size, reduction): method forward (line 23) | def forward(self, x): class IDR_DCRB (line 40) | class IDR_DCRB(nn.Module): method __init__ (line 41) | def __init__(self, conv, n_feat, kernel_size, reduction): method forward (line 48) | def forward(self, x): class DAG (line 61) | class DAG(nn.Module): method __init__ (line 62) | def __init__(self, conv, n_feat, kernel_size, reduction, n_blocks): method forward (line 73) | def forward(self, x): class KDSR (line 85) | class KDSR(nn.Module): method __init__ (line 86) | def __init__(self, args, conv=common.default_conv): method forward (line 118) | def forward(self, x, k_v): class KD_IDE (line 142) | class KD_IDE(nn.Module): method __init__ (line 143) | def __init__(self,args): method forward (line 181) | def forward(self, x): class BlindSR (line 190) | class BlindSR(nn.Module): method __init__ (line 191) | def __init__(self, args): method forward (line 200) | def forward(self, x, deg_repre): FILE: KDSR-classic/model_TA/common.py function default_conv (line 7) | def default_conv(in_channels, out_channels, kernel_size, bias=True): class MeanShift (line 11) | class MeanShift(nn.Conv2d): method __init__ (line 12) | def __init__(self, rgb_range, rgb_mean, rgb_std, sign=-1): class Upsampler (line 23) | class Upsampler(nn.Sequential): method __init__ (line 24) | def __init__(self, conv, scale, n_feat, act=False, bias=True): FILE: KDSR-classic/template.py function set_template (line 1) | def set_template(args): FILE: KDSR-classic/trainer_anisonoise_stage3.py class Trainer (line 15) | class Trainer(): method __init__ (line 16) | def __init__(self, args, loader, model_TA, my_loss, ckp): method train (line 36) | def train(self): method test (line 118) | def test(self): method crop_border (line 193) | def crop_border(self, img_hr, scale): method get_patch (line 200) | def get_patch(self, img, patch_size=48, scale=4): method crop (line 208) | def crop(self, img_hr): method terminate (line 216) | def terminate(self): FILE: KDSR-classic/trainer_anisonoise_stage4.py class Trainer (line 11) | class Trainer(): method __init__ (line 12) | def __init__(self, args, loader, model_ST, model_TA, my_loss, ckp): method train (line 37) | def train(self): method test (line 152) | def test(self): method crop_border (line 222) | def crop_border(self, img_hr, scale): method terminate (line 229) | def terminate(self): FILE: KDSR-classic/trainer_iso_stage3.py class Trainer (line 16) | class Trainer(): method __init__ (line 17) | def __init__(self, args, loader, model_TA, my_loss, ckp): method train (line 37) | def train(self): method test (line 119) | def test(self): method crop_border (line 197) | def crop_border(self, img_hr, scale): method terminate (line 204) | def terminate(self): FILE: KDSR-classic/trainer_iso_stage4.py class Trainer (line 11) | class Trainer(): method __init__ (line 12) | def __init__(self, args, loader, model_ST, model_TA, my_loss, ckp): method train (line 38) | def train(self): method test (line 153) | def test(self): method crop_border (line 230) | def crop_border(self, img_hr, scale): method terminate (line 237) | def terminate(self): FILE: KDSR-classic/utility.py class AverageMeter (line 15) | class AverageMeter(object): method __init__ (line 17) | def __init__(self): method reset (line 20) | def reset(self): method update (line 26) | def update(self, val, n=1): class timer (line 33) | class timer(): method __init__ (line 34) | def __init__(self): method tic (line 38) | def tic(self): method toc (line 41) | def toc(self): method hold (line 44) | def hold(self): method release (line 47) | def release(self): method reset (line 53) | def reset(self): class checkpoint (line 57) | class checkpoint(): method __init__ (line 58) | def __init__(self, args): method save (line 85) | def save(self, trainer, epoch, is_best=False): method add_log (line 97) | def add_log(self, log): method write_log (line 100) | def write_log(self, log, refresh=False): method done (line 107) | def done(self): method plot_psnr (line 110) | def plot_psnr(self, epoch): method save_results (line 128) | def save_results(self, filename, save_list, scale): function quantize (line 136) | def quantize(img, rgb_range): function calc_psnr (line 141) | def calc_psnr(sr, hr, scale, rgb_range, benchmark=False): function calc_ssim (line 162) | def calc_ssim(img1, img2, scale=2, benchmark=False): function ssim (line 199) | def ssim(img1, img2): function make_optimizer (line 222) | def make_optimizer(args, my_model): function make_scheduler (line 243) | def make_scheduler(args, my_optimizer): FILE: KDSR-classic/utility2.py class AverageMeter (line 14) | class AverageMeter(object): method __init__ (line 16) | def __init__(self): method reset (line 19) | def reset(self): method update (line 25) | def update(self, val, n=1): class timer (line 32) | class timer(): method __init__ (line 33) | def __init__(self): method tic (line 37) | def tic(self): method toc (line 40) | def toc(self): method hold (line 43) | def hold(self): method release (line 46) | def release(self): method reset (line 52) | def reset(self): class checkpoint (line 56) | class checkpoint(): method __init__ (line 57) | def __init__(self, args): method save (line 83) | def save(self, trainer, epoch, is_best=False): method add_log (line 95) | def add_log(self, log): method write_log (line 98) | def write_log(self, log, refresh=False): method done (line 105) | def done(self): method plot_psnr (line 108) | def plot_psnr(self, epoch): method save_results (line 126) | def save_results(self, filename, save_list, scale): function quantize (line 134) | def quantize(img, rgb_range): function calc_psnr (line 139) | def calc_psnr(sr, hr, scale, rgb_range, benchmark=False): function calc_ssim (line 160) | def calc_ssim(img1, img2, scale=2, benchmark=False): function ssim (line 197) | def ssim(img1, img2): function make_optimizer (line 220) | def make_optimizer(args, my_model): function make_scheduler (line 241) | def make_scheduler(args, my_optimizer): FILE: KDSR-classic/utility3.py function tensor2img (line 13) | def tensor2img(tensor, out_type=np.float32, min_max=(0, 1)): function bgr2ycbcr (line 42) | def bgr2ycbcr(img, only_y=True): function calc_psnr (line 66) | def calc_psnr(img1, img2,scale): function ssim (line 82) | def ssim(img1, img2): function calculate_ssim (line 106) | def calculate_ssim(img1, img2): FILE: KDSR-classic/utils/util.py function cal_sigma (line 9) | def cal_sigma(sig_x, sig_y, radians): function anisotropic_gaussian_kernel (line 22) | def anisotropic_gaussian_kernel(batch, kernel_size, covar): function isotropic_gaussian_kernel (line 35) | def isotropic_gaussian_kernel(batch, kernel_size, sigma): function random_anisotropic_gaussian_kernel (line 44) | def random_anisotropic_gaussian_kernel(batch=1, kernel_size=21, lambda_m... function stable_anisotropic_gaussian_kernel (line 54) | def stable_anisotropic_gaussian_kernel(kernel_size=21, theta=0, lambda_1... function random_isotropic_gaussian_kernel (line 64) | def random_isotropic_gaussian_kernel(batch=1, kernel_size=21, sig_min=0.... function stable_isotropic_gaussian_kernel (line 70) | def stable_isotropic_gaussian_kernel(kernel_size=21, sig=4.0): function random_gaussian_kernel (line 76) | def random_gaussian_kernel(batch, kernel_size=21, blur_type='iso_gaussia... function stable_gaussian_kernel (line 83) | def stable_gaussian_kernel(kernel_size=21, blur_type='iso_gaussian', sig... class bicubic (line 91) | class bicubic(nn.Module): method __init__ (line 92) | def __init__(self): method cubic (line 95) | def cubic(self, x): method contribute (line 106) | def contribute(self, in_size, out_size, scale): method forward (line 151) | def forward(self, input, scale=1/4): class Gaussin_Kernel (line 171) | class Gaussin_Kernel(object): method __init__ (line 172) | def __init__(self, kernel_size=21, blur_type='iso_gaussian', method __call__ (line 188) | def __call__(self, batch, random): class BatchBlur (line 201) | class BatchBlur(nn.Module): method __init__ (line 202) | def __init__(self, kernel_size=21): method forward (line 210) | def forward(self, input, kernel): class SRMDPreprocessing (line 228) | class SRMDPreprocessing(object): method __init__ (line 229) | def __init__(self, method __call__ (line 274) | def __call__(self, hr_tensor, random=True): class BicubicPreprocessing (line 313) | class BicubicPreprocessing(object): method __init__ (line 314) | def __init__(self, method __call__ (line 337) | def __call__(self, hr_tensor, random=True): FILE: KDSR-classic/utils/util2.py function cal_sigma (line 9) | def cal_sigma(sig_x, sig_y, radians): function anisotropic_gaussian_kernel (line 22) | def anisotropic_gaussian_kernel(batch, kernel_size, covar): function isotropic_gaussian_kernel (line 35) | def isotropic_gaussian_kernel(batch, kernel_size, sigma): function random_anisotropic_gaussian_kernel (line 44) | def random_anisotropic_gaussian_kernel(batch=1, kernel_size=21, lambda_m... function stable_anisotropic_gaussian_kernel (line 54) | def stable_anisotropic_gaussian_kernel(kernel_size=21, theta=0, lambda_1... function random_isotropic_gaussian_kernel (line 64) | def random_isotropic_gaussian_kernel(batch=1, kernel_size=21, sig_min=0.... function stable_isotropic_gaussian_kernel (line 70) | def stable_isotropic_gaussian_kernel(kernel_size=21, sig=4.0): function random_gaussian_kernel (line 76) | def random_gaussian_kernel(batch, kernel_size=21, blur_type='iso_gaussia... function stable_gaussian_kernel (line 83) | def stable_gaussian_kernel(kernel_size=21, blur_type='iso_gaussian', sig... class bicubic (line 91) | class bicubic(nn.Module): method __init__ (line 92) | def __init__(self): method cubic (line 95) | def cubic(self, x): method contribute (line 106) | def contribute(self, in_size, out_size, scale): method forward (line 151) | def forward(self, input, scale=1/4): class Gaussin_Kernel (line 171) | class Gaussin_Kernel(object): method __init__ (line 172) | def __init__(self, kernel_size=21, blur_type='iso_gaussian', method __call__ (line 188) | def __call__(self, batch, random): class BatchBlur (line 201) | class BatchBlur(nn.Module): method __init__ (line 202) | def __init__(self, kernel_size=21): method forward (line 210) | def forward(self, input, kernel): class SRMDPreprocessing (line 228) | class SRMDPreprocessing(object): method __init__ (line 229) | def __init__(self, method __call__ (line 274) | def __call__(self, hr_tensor, random=True):