basicsr API

basicsr.archs

• basicsr.archs.__init__
• basicsr.archs.arch_util
  • DCNv2Pack
    • DCNv2Pack.forward()
    • DCNv2Pack.training
  • ResidualBlockNoBN
    • ResidualBlockNoBN.forward()
    • ResidualBlockNoBN.training
  • Upsample
  • default_init_weights()
  • flow_warp()
  • make_layer()
  • modulated_deform_conv()
  • pixel_unshuffle()
  • resize_flow()
  • to_1tuple()
  • to_2tuple()
  • to_3tuple()
  • to_4tuple()
  • to_ntuple()
  • trunc_normal_()
• basicsr.archs.basicvsr_arch
  • BasicVSR
    • BasicVSR.forward()
    • BasicVSR.get_flow()
    • BasicVSR.training
  • ConvResidualBlocks
    • ConvResidualBlocks.forward()
    • ConvResidualBlocks.training
  • EDVRFeatureExtractor
    • EDVRFeatureExtractor.forward()
    • EDVRFeatureExtractor.training
  • IconVSR
    • IconVSR.forward()
    • IconVSR.get_flow()
    • IconVSR.get_keyframe_feature()
    • IconVSR.pad_spatial()
    • IconVSR.training
• basicsr.archs.basicvsrpp_arch
  • BasicVSRPlusPlus
    • BasicVSRPlusPlus.check_if_mirror_extended()
    • BasicVSRPlusPlus.compute_flow()
    • BasicVSRPlusPlus.forward()
    • BasicVSRPlusPlus.propagate()
    • BasicVSRPlusPlus.training
    • BasicVSRPlusPlus.upsample()
  • SecondOrderDeformableAlignment
    • SecondOrderDeformableAlignment.forward()
    • SecondOrderDeformableAlignment.init_offset()
    • SecondOrderDeformableAlignment.training
• basicsr.archs.dfdnet_arch
  • DFDNet
    • DFDNet.forward()
    • DFDNet.put_dict_to_device()
    • DFDNet.swap_feat()
    • DFDNet.training
  • SFTUpBlock
    • SFTUpBlock.forward()
    • SFTUpBlock.training
• basicsr.archs.dfdnet_util
  • AttentionBlock()
  • Blur
    • Blur.forward()
    • Blur.training
  • BlurFunction
    • BlurFunction.backward()
    • BlurFunction.forward()
  • BlurFunctionBackward
    • BlurFunctionBackward.backward()
    • BlurFunctionBackward.forward()
  • MSDilationBlock
    • MSDilationBlock.forward()
    • MSDilationBlock.training
  • UpResBlock
    • UpResBlock.forward()
    • UpResBlock.training
  • adaptive_instance_normalization()
  • blur()
  • calc_mean_std()
  • conv_block()
• basicsr.archs.discriminator_arch
  • UNetDiscriminatorSN
    • UNetDiscriminatorSN.forward()
    • UNetDiscriminatorSN.training
  • VGGStyleDiscriminator
    • VGGStyleDiscriminator.forward()
    • VGGStyleDiscriminator.training
• basicsr.archs.duf_arch
  • DUF
    • DUF.forward()
    • DUF.training
  • DenseBlocks
    • DenseBlocks.forward()
    • DenseBlocks.training
  • DenseBlocksTemporalReduce
    • DenseBlocksTemporalReduce.forward()
    • DenseBlocksTemporalReduce.training
  • DynamicUpsamplingFilter
    • DynamicUpsamplingFilter.forward()
    • DynamicUpsamplingFilter.training
• basicsr.archs.ecbsr_arch
  • ECB
    • ECB.forward()
    • ECB.rep_params()
    • ECB.training
  • ECBSR
    • ECBSR.forward()
    • ECBSR.training
  • SeqConv3x3
    • SeqConv3x3.forward()
    • SeqConv3x3.rep_params()
    • SeqConv3x3.training
• basicsr.archs.edsr_arch
  • EDSR
    • EDSR.forward()
    • EDSR.training
• basicsr.archs.edvr_arch
  • EDVR
    • EDVR.forward()
    • EDVR.training
  • PCDAlignment
    • PCDAlignment.forward()
    • PCDAlignment.training
  • PredeblurModule
    • PredeblurModule.forward()
    • PredeblurModule.training
  • TSAFusion
    • TSAFusion.forward()
    • TSAFusion.training
• basicsr.archs.hifacegan_arch
  • HiFaceGAN
    • HiFaceGAN.encode()
    • HiFaceGAN.training
  • HiFaceGANDiscriminator
    • HiFaceGANDiscriminator.downsample()
    • HiFaceGANDiscriminator.forward()
    • HiFaceGANDiscriminator.training
  • NLayerDiscriminator
    • NLayerDiscriminator.forward()
    • NLayerDiscriminator.training
  • SPADEGenerator
    • SPADEGenerator.encode()
    • SPADEGenerator.forward()
    • SPADEGenerator.mixed_guidance_forward()
    • SPADEGenerator.training
• basicsr.archs.hifacegan_util
  • BaseNetwork
    • BaseNetwork.forward()
    • BaseNetwork.init_weights()
    • BaseNetwork.training
  • LIPEncoder
    • LIPEncoder.forward()
    • LIPEncoder.training
  • SPADE
    • SPADE.forward()
    • SPADE.training
  • SPADEResnetBlock
    • SPADEResnetBlock.act()
    • SPADEResnetBlock.forward()
    • SPADEResnetBlock.shortcut()
    • SPADEResnetBlock.training
  • SimplifiedLIP
    • SimplifiedLIP.forward()
    • SimplifiedLIP.init_layer()
    • SimplifiedLIP.training
  • SoftGate
    • SoftGate.COEFF
    • SoftGate.forward()
    • SoftGate.training
  • get_nonspade_norm_layer()
  • lip2d()
• basicsr.archs.inception
  • FIDInceptionA
    • FIDInceptionA.forward()
    • FIDInceptionA.training
  • FIDInceptionC
    • FIDInceptionC.forward()
    • FIDInceptionC.training
  • FIDInceptionE_1
    • FIDInceptionE_1.forward()
    • FIDInceptionE_1.training
  • FIDInceptionE_2
    • FIDInceptionE_2.forward()
    • FIDInceptionE_2.training
  • InceptionV3
    • InceptionV3.BLOCK_INDEX_BY_DIM
    • InceptionV3.DEFAULT_BLOCK_INDEX
    • InceptionV3.forward()
    • InceptionV3.training
  • fid_inception_v3()
• basicsr.archs.rcan_arch
  • ChannelAttention
    • ChannelAttention.forward()
    • ChannelAttention.training
  • RCAB
    • RCAB.forward()
    • RCAB.training
  • RCAN
    • RCAN.forward()
    • RCAN.training
  • ResidualGroup
    • ResidualGroup.forward()
    • ResidualGroup.training
• basicsr.archs.ridnet_arch
  • ChannelAttention
    • ChannelAttention.forward()
    • ChannelAttention.training
  • EAM
    • EAM.merge
    • EAM.block1
    • EAM.block2
    • EAM.ca
    • EAM.forward()
    • EAM.training
  • EResidualBlockNoBN
    • EResidualBlockNoBN.forward()
    • EResidualBlockNoBN.training
  • MeanShift
    • MeanShift.bias
    • MeanShift.dilation
    • MeanShift.groups
    • MeanShift.in_channels
    • MeanShift.kernel_size
    • MeanShift.out_channels
    • MeanShift.output_padding
    • MeanShift.padding
    • MeanShift.padding_mode
    • MeanShift.stride
    • MeanShift.transposed
    • MeanShift.weight
  • MergeRun
    • MergeRun.forward()
    • MergeRun.training
  • RIDNet
    • RIDNet.forward()
    • RIDNet.training
• basicsr.archs.rrdbnet_arch
  • RRDB
    • RRDB.forward()
    • RRDB.training
  • RRDBNet
    • RRDBNet.forward()
    • RRDBNet.training
  • ResidualDenseBlock
    • ResidualDenseBlock.forward()
    • ResidualDenseBlock.training
• basicsr.archs.spynet_arch
  • BasicModule
    • BasicModule.forward()
    • BasicModule.training
  • SpyNet
    • SpyNet.forward()
    • SpyNet.preprocess()
    • SpyNet.process()
    • SpyNet.training
• basicsr.archs.srresnet_arch
  • MSRResNet
    • MSRResNet.forward()
    • MSRResNet.training
• basicsr.archs.srvgg_arch
  • SRVGGNetCompact
    • SRVGGNetCompact.forward()
    • SRVGGNetCompact.training
• basicsr.archs.stylegan2_arch
  • ConstantInput
    • ConstantInput.forward()
    • ConstantInput.training
  • ConvLayer
  • EqualConv2d
    • EqualConv2d.forward()
    • EqualConv2d.training
  • EqualLinear
    • EqualLinear.forward()
    • EqualLinear.training
  • ModulatedConv2d
    • ModulatedConv2d.forward()
    • ModulatedConv2d.training
  • NormStyleCode
    • NormStyleCode.forward()
    • NormStyleCode.training
  • ResBlock
    • ResBlock.forward()
    • ResBlock.training
  • ScaledLeakyReLU
    • ScaledLeakyReLU.forward()
    • ScaledLeakyReLU.training
  • StyleConv
    • StyleConv.forward()
    • StyleConv.training
  • StyleGAN2Discriminator
    • StyleGAN2Discriminator.forward()
    • StyleGAN2Discriminator.training
  • StyleGAN2Generator
    • StyleGAN2Generator.forward()
    • StyleGAN2Generator.get_latent()
    • StyleGAN2Generator.make_noise()
    • StyleGAN2Generator.mean_latent()
    • StyleGAN2Generator.training
  • ToRGB
    • ToRGB.forward()
    • ToRGB.training
  • UpFirDnDownsample
    • UpFirDnDownsample.forward()
    • UpFirDnDownsample.training
  • UpFirDnSmooth
    • UpFirDnSmooth.forward()
    • UpFirDnSmooth.training
  • UpFirDnUpsample
    • UpFirDnUpsample.forward()
    • UpFirDnUpsample.training
  • make_resample_kernel()
• basicsr.archs.stylegan2_bilinear_arch
  • ConstantInput
    • ConstantInput.forward()
    • ConstantInput.training
  • ConvLayer
  • EqualConv2d
    • EqualConv2d.forward()
    • EqualConv2d.training
  • EqualLinear
    • EqualLinear.forward()
    • EqualLinear.training
  • ModulatedConv2d
    • ModulatedConv2d.forward()
    • ModulatedConv2d.training
  • NormStyleCode
    • NormStyleCode.forward()
    • NormStyleCode.training
  • ResBlock
    • ResBlock.forward()
    • ResBlock.training
  • ScaledLeakyReLU
    • ScaledLeakyReLU.forward()
    • ScaledLeakyReLU.training
  • StyleConv
    • StyleConv.forward()
    • StyleConv.training
  • StyleGAN2GeneratorBilinear
    • StyleGAN2GeneratorBilinear.forward()
    • StyleGAN2GeneratorBilinear.get_latent()
    • StyleGAN2GeneratorBilinear.make_noise()
    • StyleGAN2GeneratorBilinear.mean_latent()
    • StyleGAN2GeneratorBilinear.training
  • ToRGB
    • ToRGB.forward()
    • ToRGB.training
• basicsr.archs.swinir_arch
  • BasicLayer
    • BasicLayer.extra_repr()
    • BasicLayer.flops()
    • BasicLayer.forward()
    • BasicLayer.training
  • DropPath
    • DropPath.forward()
    • DropPath.training
  • Mlp
    • Mlp.forward()
    • Mlp.training
  • PatchEmbed
    • PatchEmbed.flops()
    • PatchEmbed.forward()
    • PatchEmbed.training
  • PatchMerging
    • PatchMerging.extra_repr()
    • PatchMerging.flops()
    • PatchMerging.forward()
    • PatchMerging.training
  • PatchUnEmbed
    • PatchUnEmbed.flops()
    • PatchUnEmbed.forward()
    • PatchUnEmbed.training
  • RSTB
    • RSTB.flops()
    • RSTB.forward()
    • RSTB.training
  • SwinIR
    • SwinIR.flops()
    • SwinIR.forward()
    • SwinIR.forward_features()
    • SwinIR.no_weight_decay()
    • SwinIR.no_weight_decay_keywords()
    • SwinIR.training
  • SwinTransformerBlock
    • SwinTransformerBlock.calculate_mask()
    • SwinTransformerBlock.extra_repr()
    • SwinTransformerBlock.flops()
    • SwinTransformerBlock.forward()
    • SwinTransformerBlock.training
  • Upsample
  • UpsampleOneStep
    • UpsampleOneStep.flops()
  • WindowAttention
    • WindowAttention.extra_repr()
    • WindowAttention.flops()
    • WindowAttention.forward()
    • WindowAttention.training
  • drop_path()
  • window_partition()
  • window_reverse()
• basicsr.archs.tof_arch
  • BasicModule
    • BasicModule.forward()
    • BasicModule.training
  • SPyNetTOF
    • SPyNetTOF.forward()
    • SPyNetTOF.training
  • TOFlow
    • TOFlow.denormalize()
    • TOFlow.forward()
    • TOFlow.normalize()
    • TOFlow.training
• basicsr.archs.vgg_arch
  • VGGFeatureExtractor
    • VGGFeatureExtractor.forward()
    • VGGFeatureExtractor.training
  • insert_bn()

basicsr.data

• basicsr.data.__init__
  • build_dataloader()
  • build_dataset()
• basicsr.data.data_sampler
  • EnlargedSampler
    • EnlargedSampler.set_epoch()
• basicsr.data.data_util
  • duf_downsample()
  • generate_frame_indices()
  • generate_gaussian_kernel()
  • paired_paths_from_folder()
  • paired_paths_from_lmdb()
  • paired_paths_from_meta_info_file()
  • paths_from_folder()
  • paths_from_lmdb()
  • read_img_seq()
• basicsr.data.degradations
  • add_gaussian_noise()
  • add_gaussian_noise_pt()
  • add_jpg_compression()
  • add_poisson_noise()
  • add_poisson_noise_pt()
  • bivariate_Gaussian()
  • bivariate_generalized_Gaussian()
  • bivariate_plateau()
  • cdf2()
  • circular_lowpass_kernel()
  • generate_gaussian_noise()
  • generate_gaussian_noise_pt()
  • generate_poisson_noise()
  • generate_poisson_noise_pt()
  • mesh_grid()
  • pdf2()
  • random_add_gaussian_noise()
  • random_add_gaussian_noise_pt()
  • random_add_jpg_compression()
  • random_add_poisson_noise()
  • random_add_poisson_noise_pt()
  • random_bivariate_Gaussian()
  • random_bivariate_generalized_Gaussian()
  • random_bivariate_plateau()
  • random_generate_gaussian_noise()
  • random_generate_gaussian_noise_pt()
  • random_generate_poisson_noise()
  • random_generate_poisson_noise_pt()
  • random_mixed_kernels()
  • sigma_matrix2()
• basicsr.data.ffhq_dataset
  • FFHQDataset
• basicsr.data.paired_image_dataset
  • PairedImageDataset
• basicsr.data.prefetch_dataloader
  • CPUPrefetcher
    • CPUPrefetcher.next()
    • CPUPrefetcher.reset()
  • CUDAPrefetcher
    • CUDAPrefetcher.next()
    • CUDAPrefetcher.preload()
    • CUDAPrefetcher.reset()
  • PrefetchDataLoader
    • PrefetchDataLoader.batch_size
    • PrefetchDataLoader.dataset
    • PrefetchDataLoader.drop_last
    • PrefetchDataLoader.num_workers
    • PrefetchDataLoader.pin_memory
    • PrefetchDataLoader.pin_memory_device
    • PrefetchDataLoader.prefetch_factor
    • PrefetchDataLoader.sampler
    • PrefetchDataLoader.timeout
  • PrefetchGenerator
    • PrefetchGenerator.run()
• basicsr.data.realesrgan_dataset
  • RealESRGANDataset
• basicsr.data.realesrgan_paired_dataset
  • RealESRGANPairedDataset
• basicsr.data.reds_dataset
  • REDSDataset
  • REDSRecurrentDataset
• basicsr.data.single_image_dataset
  • SingleImageDataset
• basicsr.data.transforms
  • augment()
  • img_rotate()
  • mod_crop()
  • paired_random_crop()
• basicsr.data.video_test_dataset
  • VideoRecurrentTestDataset
  • VideoTestDUFDataset
  • VideoTestDataset
  • VideoTestVimeo90KDataset
• basicsr.data.vimeo90k_dataset
  • Vimeo90KDataset
  • Vimeo90KRecurrentDataset

basicsr.losses

• basicsr.losses.__init__
  • build_loss()
  • g_path_regularize()
  • gradient_penalty_loss()
  • r1_penalty()
• basicsr.losses.basic_loss
  • CharbonnierLoss
    • CharbonnierLoss.forward()
    • CharbonnierLoss.training
  • L1Loss
    • L1Loss.forward()
    • L1Loss.training
  • MSELoss
    • MSELoss.forward()
    • MSELoss.training
  • PerceptualLoss
    • PerceptualLoss.forward()
    • PerceptualLoss.training
  • WeightedTVLoss
    • WeightedTVLoss.forward()
    • WeightedTVLoss.training
  • charbonnier_loss()
  • l1_loss()
  • mse_loss()
• basicsr.losses.gan_loss
  • GANLoss
    • GANLoss.forward()
    • GANLoss.get_target_label()
    • GANLoss.training
  • MultiScaleGANLoss
    • MultiScaleGANLoss.forward()
    • MultiScaleGANLoss.training
  • g_path_regularize()
  • gradient_penalty_loss()
  • r1_penalty()
• basicsr.losses.loss_util
  • get_local_weights()
  • get_refined_artifact_map()
  • reduce_loss()
  • weight_reduce_loss()
  • weighted_loss()

basicsr.metrics

• basicsr.metrics.__init__
  • calculate_niqe()
  • calculate_psnr()
  • calculate_ssim()
• basicsr.metrics.fid
  • calculate_fid()
  • extract_inception_features()
  • load_patched_inception_v3()
• basicsr.metrics.metric_util
  • reorder_image()
  • to_y_channel()
• basicsr.metrics.niqe
  • calculate_niqe()
  • compute_feature()
  • estimate_aggd_param()
  • niqe()
• basicsr.metrics.psnr_ssim
  • calculate_psnr()
  • calculate_psnr_pt()
  • calculate_ssim()
  • calculate_ssim_pt()
• basicsr.metrics.test_metrics.test_psnr_ssim
  • test()

basicsr.models

• basicsr.models.__init__
  • build_model()
• basicsr.models.base_model
  • BaseModel
    • BaseModel.feed_data()
    • BaseModel.get_bare_model()
    • BaseModel.get_current_learning_rate()
    • BaseModel.get_current_log()
    • BaseModel.get_current_visuals()
    • BaseModel.get_optimizer()
    • BaseModel.load_network()
    • BaseModel.model_ema()
    • BaseModel.model_to_device()
    • BaseModel.optimize_parameters()
    • BaseModel.print_network()
    • BaseModel.reduce_loss_dict()
    • BaseModel.resume_training()
    • BaseModel.save()
    • BaseModel.save_network()
    • BaseModel.save_training_state()
    • BaseModel.setup_schedulers()
    • BaseModel.update_learning_rate()
    • BaseModel.validation()
• basicsr.models.edvr_model
  • EDVRModel
    • EDVRModel.optimize_parameters()
    • EDVRModel.setup_optimizers()
• basicsr.models.esrgan_model
  • ESRGANModel
    • ESRGANModel.optimize_parameters()
• basicsr.models.hifacegan_model
  • HiFaceGANModel
    • HiFaceGANModel.discriminate()
    • HiFaceGANModel.init_training_settings()
    • HiFaceGANModel.nondist_validation()
    • HiFaceGANModel.optimize_parameters()
    • HiFaceGANModel.save()
    • HiFaceGANModel.setup_optimizers()
    • HiFaceGANModel.validation()
• basicsr.models.lr_scheduler
  • CosineAnnealingRestartLR
    • CosineAnnealingRestartLR.get_lr()
  • MultiStepRestartLR
    • MultiStepRestartLR.get_lr()
  • get_position_from_periods()
• basicsr.models.realesrgan_model
  • RealESRGANModel
    • RealESRGANModel.feed_data()
    • RealESRGANModel.nondist_validation()
    • RealESRGANModel.optimize_parameters()
• basicsr.models.realesrnet_model
  • RealESRNetModel
    • RealESRNetModel.feed_data()
    • RealESRNetModel.nondist_validation()
• basicsr.models.sr_model
  • SRModel
    • SRModel.dist_validation()
    • SRModel.feed_data()
    • SRModel.get_current_visuals()
    • SRModel.init_training_settings()
    • SRModel.nondist_validation()
    • SRModel.optimize_parameters()
    • SRModel.save()
    • SRModel.setup_optimizers()
    • SRModel.test()
    • SRModel.test_selfensemble()
• basicsr.models.srgan_model
  • SRGANModel
    • SRGANModel.init_training_settings()
    • SRGANModel.optimize_parameters()
    • SRGANModel.save()
    • SRGANModel.setup_optimizers()
• basicsr.models.stylegan2_model
  • StyleGAN2Model
    • StyleGAN2Model.dist_validation()
    • StyleGAN2Model.feed_data()
    • StyleGAN2Model.init_training_settings()
    • StyleGAN2Model.make_noise()
    • StyleGAN2Model.mixing_noise()
    • StyleGAN2Model.nondist_validation()
    • StyleGAN2Model.optimize_parameters()
    • StyleGAN2Model.save()
    • StyleGAN2Model.setup_optimizers()
    • StyleGAN2Model.test()
• basicsr.models.swinir_model
  • SwinIRModel
    • SwinIRModel.test()
• basicsr.models.video_base_model
  • VideoBaseModel
    • VideoBaseModel.dist_validation()
    • VideoBaseModel.nondist_validation()
• basicsr.models.video_gan_model
  • VideoGANModel
• basicsr.models.video_recurrent_gan_model
  • VideoRecurrentGANModel
    • VideoRecurrentGANModel.init_training_settings()
    • VideoRecurrentGANModel.optimize_parameters()
    • VideoRecurrentGANModel.save()
    • VideoRecurrentGANModel.setup_optimizers()
• basicsr.models.video_recurrent_model
  • VideoRecurrentModel
    • VideoRecurrentModel.dist_validation()
    • VideoRecurrentModel.optimize_parameters()
    • VideoRecurrentModel.setup_optimizers()
    • VideoRecurrentModel.test()

basicsr.ops

• basicsr.ops.__init__
• basicsr.ops.dcn.__init__
  • DeformConv
    • DeformConv.forward()
    • DeformConv.reset_parameters()
    • DeformConv.training
  • DeformConvPack
    • DeformConvPack.forward()
    • DeformConvPack.init_offset()
    • DeformConvPack.training
  • ModulatedDeformConv
    • ModulatedDeformConv.forward()
    • ModulatedDeformConv.init_weights()
    • ModulatedDeformConv.training
  • ModulatedDeformConvPack
    • ModulatedDeformConvPack.forward()
    • ModulatedDeformConvPack.init_weights()
    • ModulatedDeformConvPack.training
  • deform_conv()
  • modulated_deform_conv()
• basicsr.ops.dcn.deform_conv
  • DeformConv
    • DeformConv.forward()
    • DeformConv.reset_parameters()
    • DeformConv.training
  • DeformConvFunction
    • DeformConvFunction.backward()
    • DeformConvFunction.forward()
  • DeformConvPack
    • DeformConvPack.forward()
    • DeformConvPack.init_offset()
    • DeformConvPack.training
  • ModulatedDeformConv
    • ModulatedDeformConv.forward()
    • ModulatedDeformConv.init_weights()
    • ModulatedDeformConv.training
  • ModulatedDeformConvFunction
    • ModulatedDeformConvFunction.backward()
    • ModulatedDeformConvFunction.forward()
  • ModulatedDeformConvPack
    • ModulatedDeformConvPack.forward()
    • ModulatedDeformConvPack.init_weights()
    • ModulatedDeformConvPack.training
  • deform_conv()
  • modulated_deform_conv()
• basicsr.ops.fused_act.__init__
  • FusedLeakyReLU
    • FusedLeakyReLU.forward()
    • FusedLeakyReLU.training
  • fused_leaky_relu()
• basicsr.ops.fused_act.fused_act
  • FusedLeakyReLU
    • FusedLeakyReLU.forward()
    • FusedLeakyReLU.training
  • FusedLeakyReLUFunction
    • FusedLeakyReLUFunction.backward()
    • FusedLeakyReLUFunction.forward()
  • FusedLeakyReLUFunctionBackward
    • FusedLeakyReLUFunctionBackward.backward()
    • FusedLeakyReLUFunctionBackward.forward()
  • fused_leaky_relu()
• basicsr.ops.upfirdn2d.__init__
  • upfirdn2d()
• basicsr.ops.upfirdn2d.upfirdn2d
  • UpFirDn2d
    • UpFirDn2d.backward()
    • UpFirDn2d.forward()
  • UpFirDn2dBackward
    • UpFirDn2dBackward.backward()
    • UpFirDn2dBackward.forward()
  • upfirdn2d()
  • upfirdn2d_native()

basicsr.test

• basicsr.test
  • test_pipeline()

basicsr.train

• basicsr.train
  • create_train_val_dataloader()
  • init_tb_loggers()
  • load_resume_state()
  • train_pipeline()

basicsr.utils

• basicsr.utils.__init__
  • AvgTimer
    • AvgTimer.get_avg_time()
    • AvgTimer.get_current_time()
    • AvgTimer.record()
    • AvgTimer.start()
  • DiffJPEG
    • DiffJPEG.forward()
    • DiffJPEG.training
  • FileClient
    • FileClient.backend
    • FileClient.client
    • FileClient.get()
    • FileClient.get_text()
  • MessageLogger
    • MessageLogger.reset_start_time()
  • USMSharp
    • USMSharp.forward()
    • USMSharp.training
  • bgr2ycbcr()
  • check_resume()
  • crop_border()
  • get_env_info()
  • get_root_logger()
  • get_time_str()
  • imfrombytes()
  • img2tensor()
  • imwrite()
  • init_tb_logger()
  • init_wandb_logger()
  • make_exp_dirs()
  • mkdir_and_rename()
  • rgb2ycbcr()
  • rgb2ycbcr_pt()
  • scandir()
  • set_random_seed()
  • sizeof_fmt()
  • tensor2img()
  • usm_sharp()
  • yaml_load()
  • ycbcr2bgr()
  • ycbcr2rgb()
• basicsr.utils.color_util
  • bgr2ycbcr()
  • rgb2ycbcr()
  • rgb2ycbcr_pt()
  • ycbcr2bgr()
  • ycbcr2rgb()
• basicsr.utils.diffjpeg
  • BlockMerging
    • BlockMerging.forward()
    • BlockMerging.training
  • BlockSplitting
    • BlockSplitting.forward()
    • BlockSplitting.training
  • CDequantize
    • CDequantize.forward()
    • CDequantize.training
  • CQuantize
    • CQuantize.forward()
    • CQuantize.training
  • ChromaSubsampling
    • ChromaSubsampling.forward()
    • ChromaSubsampling.training
  • ChromaUpsampling
    • ChromaUpsampling.forward()
    • ChromaUpsampling.training
  • CompressJpeg
    • CompressJpeg.forward()
    • CompressJpeg.training
  • DCT8x8
    • DCT8x8.forward()
    • DCT8x8.training
  • DeCompressJpeg
    • DeCompressJpeg.forward()
    • DeCompressJpeg.training
  • DiffJPEG
    • DiffJPEG.forward()
    • DiffJPEG.training
  • RGB2YCbCrJpeg
    • RGB2YCbCrJpeg.forward()
    • RGB2YCbCrJpeg.training
  • YCbCr2RGBJpeg
    • YCbCr2RGBJpeg.forward()
    • YCbCr2RGBJpeg.training
  • YDequantize
    • YDequantize.forward()
    • YDequantize.training
  • YQuantize
    • YQuantize.forward()
    • YQuantize.training
  • diff_round()
  • iDCT8x8
    • iDCT8x8.forward()
    • iDCT8x8.training
  • quality_to_factor()
• basicsr.utils.dist_util
  • get_dist_info()
  • init_dist()
  • master_only()
• basicsr.utils.download_util
  • download_file_from_google_drive()
  • get_confirm_token()
  • load_file_from_url()
  • save_response_content()
• basicsr.utils.file_client
  • BaseStorageBackend
    • BaseStorageBackend.get()
    • BaseStorageBackend.get_text()
  • FileClient
    • FileClient.backend
    • FileClient.client
    • FileClient.get()
    • FileClient.get_text()
  • HardDiskBackend
    • HardDiskBackend.get()
    • HardDiskBackend.get_text()
  • LmdbBackend
    • LmdbBackend.db_paths
    • LmdbBackend._client
    • LmdbBackend.get()
    • LmdbBackend.get_text()
  • MemcachedBackend
    • MemcachedBackend.server_list_cfg
    • MemcachedBackend.client_cfg
    • MemcachedBackend.sys_path
    • MemcachedBackend.get()
    • MemcachedBackend.get_text()
• basicsr.utils.flow_util
  • dequantize()
  • dequantize_flow()
  • flowread()
  • flowwrite()
  • quantize()
  • quantize_flow()
• basicsr.utils.img_process_util
  • USMSharp
    • USMSharp.forward()
    • USMSharp.training
  • filter2D()
  • usm_sharp()
• basicsr.utils.img_util
  • crop_border()
  • imfrombytes()
  • img2tensor()
  • imwrite()
  • tensor2img()
  • tensor2img_fast()
• basicsr.utils.lmdb_util
  • LmdbMaker
    • LmdbMaker.close()
    • LmdbMaker.put()
  • make_lmdb_from_imgs()
  • read_img_worker()
• basicsr.utils.logger
  • AvgTimer
    • AvgTimer.get_avg_time()
    • AvgTimer.get_current_time()
    • AvgTimer.record()
    • AvgTimer.start()
  • MessageLogger
    • MessageLogger.reset_start_time()
  • get_env_info()
  • get_root_logger()
  • init_tb_logger()
  • init_wandb_logger()
• basicsr.utils.matlab_functions
  • calculate_weights_indices()
  • cubic()
  • imresize()
• basicsr.utils.misc
  • check_resume()
  • get_time_str()
  • make_exp_dirs()
  • mkdir_and_rename()
  • scandir()
  • set_random_seed()
  • sizeof_fmt()
• basicsr.utils.options
  • copy_opt_file()
  • dict2str()
  • ordered_yaml()
  • parse_options()
  • yaml_load()
• basicsr.utils.plot_util
  • read_data_from_tensorboard()
  • read_data_from_txt_1v()
  • read_data_from_txt_2v()
  • smooth_data()
• basicsr.utils.registry
  • Registry
    • Registry.get()
    • Registry.keys()
    • Registry.register()