MMEditing
latest
Get Started
Prerequisites
Installation
Demo
Model Zoo
Overview
Inpainting Models
Matting Models
Super-Resolution Models
Generation Models
Datasets
Overview
Inpainting Datasets
Matting Datasets
Super-Resolution Datasets
Generation Datasets
Quick Run
Inference with pretrained models
Train a model
Tutorials
Tutorial 1: Learn about Configs
Useful Tools and Scripts
Useful tools
Notes
Benchmark
FAQ
FAQ
API Reference
API Reference
MMEditing
»
Index
Edit on GitHub
Index
A
|
B
|
C
|
D
|
E
|
F
|
G
|
H
|
I
|
L
|
M
|
N
|
O
|
P
|
Q
|
R
|
S
|
T
|
U
|
V
|
W
A
AdobeComp1kDataset (class in mmedit.datasets)
after_train_iter() (mmedit.core.DistEvalIterHook method)
(mmedit.core.EvalIterHook method)
(mmedit.core.VisualizationHook method)
ASPP (class in mmedit.models.common)
B
backward_discriminator() (mmedit.models.Pix2Pix method)
backward_discriminators() (mmedit.models.CycleGAN method)
backward_generator() (mmedit.models.Pix2Pix method)
backward_generators() (mmedit.models.CycleGAN method)
BaseDataset (class in mmedit.datasets)
BaseGenerationDataset (class in mmedit.datasets)
BaseMattingDataset (class in mmedit.datasets)
BaseMattor (class in mmedit.models)
BaseModel (class in mmedit.models)
BaseSRDataset (class in mmedit.datasets)
BasicRestorer (class in mmedit.models)
BinarizeImage (class in mmedit.datasets.pipelines)
build() (in module mmedit.models)
build_backbone() (in module mmedit.models)
build_component() (in module mmedit.models)
build_dataloader() (in module mmedit.datasets)
build_dataset() (in module mmedit.datasets)
build_loss() (in module mmedit.models)
build_model() (in module mmedit.models)
build_optimizers() (in module mmedit.core)
C
calculate_loss_with_type() (mmedit.models.DeepFillv1Inpaintor method)
(mmedit.models.TwoStageInpaintor method)
calculate_overlap_factor() (mmedit.models.common.ContextualAttentionModule method)
calculate_unfold_hw() (mmedit.models.common.ContextualAttentionModule method)
CharbonnierCompLoss (class in mmedit.models.losses)
CharbonnierLoss (class in mmedit.models.losses)
Collect (class in mmedit.datasets.pipelines)
Compose (class in mmedit.datasets.pipelines)
CompositeFg (class in mmedit.datasets.pipelines)
compute_guided_attention_score() (mmedit.models.common.GCAModule method)
compute_similarity_map() (mmedit.models.common.GCAModule method)
ContextualAttentionModule (class in mmedit.models.common)
ContextualAttentionNeck (class in mmedit.models.backbones)
Crop (class in mmedit.datasets.pipelines)
CropAroundCenter (class in mmedit.datasets.pipelines)
CropAroundFg (class in mmedit.datasets.pipelines)
CropAroundUnknown (class in mmedit.datasets.pipelines)
CycleGAN (class in mmedit.models)
D
DeepFillDecoder (class in mmedit.models.backbones)
DeepFillEncoder (class in mmedit.models.backbones)
DeepFillEncoderDecoder (class in mmedit.models.backbones)
DeepFillRefiner (class in mmedit.models.components)
DeepFillv1Discriminators (class in mmedit.models.components)
DeepFillv1Inpaintor (class in mmedit.models)
default_init_weights() (in module mmedit.models.common)
denormalize() (mmedit.models.backbones.TOFlow method)
DepthwiseIndexBlock (class in mmedit.models.backbones)
DepthwiseSeparableConvModule (class in mmedit.models.common)
DIM (class in mmedit.models)
DiscShiftLoss (class in mmedit.models.losses)
DistEvalIterHook (class in mmedit.core)
E
EDSR (class in mmedit.models.backbones)
EDVRNet (class in mmedit.models.backbones)
ESRGAN (class in mmedit.models)
EvalIterHook (class in mmedit.core)
evaluate() (mmedit.core.EvalIterHook method)
(mmedit.datasets.BaseGenerationDataset method)
(mmedit.datasets.BaseMattingDataset method)
(mmedit.datasets.BaseSRDataset method)
(mmedit.models.BaseMattor method)
(mmedit.models.BasicRestorer method)
extract_around_bbox() (in module mmedit.models.common)
extract_bbox_patch() (in module mmedit.models.common)
extract_feature_maps_patches() (mmedit.models.common.GCAModule method)
extract_patches() (mmedit.models.common.GCAModule method)
F
FixedCrop (class in mmedit.datasets.pipelines)
Flip (class in mmedit.datasets.pipelines)
flow_warp() (in module mmedit.models.common)
FormatTrimap (class in mmedit.datasets.pipelines)
forward() (mmedit.models.backbones.ContextualAttentionNeck method)
(mmedit.models.backbones.DeepFillDecoder method)
(mmedit.models.backbones.DeepFillEncoder method)
(mmedit.models.backbones.DeepFillEncoderDecoder method)
(mmedit.models.backbones.DepthwiseIndexBlock method)
(mmedit.models.backbones.EDSR method)
(mmedit.models.backbones.EDVRNet method)
(mmedit.models.backbones.GLDecoder method)
(mmedit.models.backbones.GLDilationNeck method)
(mmedit.models.backbones.GLEncoder method)
(mmedit.models.backbones.GLEncoderDecoder method)
(mmedit.models.backbones.HolisticIndexBlock method)
(mmedit.models.backbones.IndexedUpsample method)
(mmedit.models.backbones.IndexNetDecoder method)
(mmedit.models.backbones.IndexNetEncoder method)
(mmedit.models.backbones.MSRResNet method)
(mmedit.models.backbones.PConvDecoder method)
(mmedit.models.backbones.PConvEncoder method)
(mmedit.models.backbones.PConvEncoderDecoder method)
(mmedit.models.backbones.PlainDecoder method)
(mmedit.models.backbones.ResGCADecoder method)
(mmedit.models.backbones.ResGCAEncoder method)
(mmedit.models.backbones.ResNetDec method)
(mmedit.models.backbones.ResNetEnc method)
(mmedit.models.backbones.ResnetGenerator method)
(mmedit.models.backbones.ResShortcutDec method)
(mmedit.models.backbones.ResShortcutEnc method)
(mmedit.models.backbones.RRDBNet method)
(mmedit.models.backbones.SimpleEncoderDecoder method)
(mmedit.models.backbones.SRCNN method)
(mmedit.models.backbones.TOFlow method)
(mmedit.models.backbones.UnetGenerator method)
(mmedit.models.backbones.VGG16 method)
(mmedit.models.BaseMattor method)
(mmedit.models.BaseModel method)
(mmedit.models.BasicRestorer method)
(mmedit.models.common.ASPP method)
(mmedit.models.common.ContextualAttentionModule method)
(mmedit.models.common.DepthwiseSeparableConvModule method)
(mmedit.models.common.GCAModule method)
(mmedit.models.common.LinearModule method)
(mmedit.models.common.MaskConvModule method)
(mmedit.models.common.PartialConv2d method)
(mmedit.models.common.PixelShufflePack method)
(mmedit.models.common.ResidualBlockNoBN method)
(mmedit.models.common.ResidualBlockWithDropout method)
(mmedit.models.common.SimpleGatedConvModule method)
(mmedit.models.common.UnetSkipConnectionBlock method)
(mmedit.models.components.DeepFillRefiner method)
(mmedit.models.components.DeepFillv1Discriminators method)
(mmedit.models.components.GLDiscs method)
(mmedit.models.components.ModifiedVGG method)
(mmedit.models.components.MultiLayerDiscriminator method)
(mmedit.models.components.PatchDiscriminator method)
(mmedit.models.components.PlainRefiner method)
(mmedit.models.CycleGAN method)
(mmedit.models.losses.CharbonnierCompLoss method)
(mmedit.models.losses.CharbonnierLoss method)
(mmedit.models.losses.DiscShiftLoss method)
(mmedit.models.losses.GANLoss method)
(mmedit.models.losses.GradientLoss method)
(mmedit.models.losses.GradientPenaltyLoss method)
(mmedit.models.losses.L1CompositionLoss method)
(mmedit.models.losses.L1Loss method)
(mmedit.models.losses.MaskedTVLoss method)
(mmedit.models.losses.MSECompositionLoss method)
(mmedit.models.losses.MSELoss method)
(mmedit.models.losses.PerceptualLoss method)
(mmedit.models.losses.PerceptualVGG method)
(mmedit.models.OneStageInpaintor method)
(mmedit.models.Pix2Pix method)
(mmedit.models.SRGAN method)
forward_dummy() (mmedit.models.BasicRestorer method)
(mmedit.models.CycleGAN method)
(mmedit.models.OneStageInpaintor method)
(mmedit.models.PConvInpaintor method)
(mmedit.models.Pix2Pix method)
forward_test() (mmedit.models.BaseMattor method)
(mmedit.models.BaseModel method)
(mmedit.models.BasicRestorer method)
(mmedit.models.CycleGAN method)
(mmedit.models.DIM method)
(mmedit.models.GCA method)
(mmedit.models.IndexNet method)
(mmedit.models.OneStageInpaintor method)
(mmedit.models.PConvInpaintor method)
(mmedit.models.Pix2Pix method)
(mmedit.models.TwoStageInpaintor method)
forward_train() (mmedit.models.BaseMattor method)
(mmedit.models.BaseModel method)
(mmedit.models.BasicRestorer method)
(mmedit.models.CycleGAN method)
(mmedit.models.DIM method)
(mmedit.models.GCA method)
(mmedit.models.IndexNet method)
(mmedit.models.OneStageInpaintor method)
(mmedit.models.Pix2Pix method)
forward_train_d() (mmedit.models.DeepFillv1Inpaintor method)
(mmedit.models.OneStageInpaintor method)
freeze_backbone() (mmedit.models.BaseMattor method)
freeze_bn() (mmedit.models.backbones.IndexNetEncoder method)
fuse_correlation_map() (mmedit.models.common.ContextualAttentionModule method)
G
GANImageBuffer (class in mmedit.models.common)
GANLoss (class in mmedit.models.losses)
GCA (class in mmedit.models)
GCAModule (class in mmedit.models.common)
GenerateFrameIndices (class in mmedit.datasets.pipelines)
GenerateFrameIndiceswithPadding (class in mmedit.datasets.pipelines)
GenerateSeg (class in mmedit.datasets.pipelines)
GenerateSoftSeg (class in mmedit.datasets.pipelines)
GenerateTrimap (class in mmedit.datasets.pipelines)
GenerateTrimapWithDistTransform (class in mmedit.datasets.pipelines)
generation_init_weights() (in module mmedit.models.common)
GenerationPairedDataset (class in mmedit.datasets)
GenerationUnpairedDataset (class in mmedit.datasets)
generator_loss() (mmedit.models.GLInpaintor method)
(mmedit.models.OneStageInpaintor method)
get_lr() (mmedit.core.LinearLrUpdaterHook method)
get_module() (mmedit.models.CycleGAN method)
(mmedit.models.DeepFillv1Inpaintor method)
get_root_logger() (in module mmedit.utils)
get_target_label() (mmedit.models.losses.GANLoss method)
GetMaskedImage (class in mmedit.datasets.pipelines)
GetSpatialDiscountMask (class in mmedit.datasets.pipelines)
GLDecoder (class in mmedit.models.backbones)
GLDilationNeck (class in mmedit.models.backbones)
GLDiscs (class in mmedit.models.components)
GLEncoder (class in mmedit.models.backbones)
GLEncoderDecoder (class in mmedit.models.backbones)
GLInpaintor (class in mmedit.models)
GradientLoss (class in mmedit.models.losses)
GradientPenaltyLoss (class in mmedit.models.losses)
H
HolisticIndexBlock (class in mmedit.models.backbones)
I
im2col() (mmedit.models.common.ContextualAttentionModule method)
ImageToTensor (class in mmedit.datasets.pipelines)
ImgInpaintingDataset (class in mmedit.datasets)
IndexedUpsample (class in mmedit.models.backbones)
IndexNet (class in mmedit.models)
IndexNetDecoder (class in mmedit.models.backbones)
IndexNetEncoder (class in mmedit.models.backbones)
init_weights() (mmedit.models.backbones.DeepFillEncoderDecoder method)
(mmedit.models.backbones.EDSR method)
(mmedit.models.backbones.EDVRNet method)
(mmedit.models.backbones.GLEncoderDecoder method)
(mmedit.models.backbones.IndexedUpsample method)
(mmedit.models.backbones.IndexNetDecoder method)
(mmedit.models.backbones.IndexNetEncoder method)
(mmedit.models.backbones.MSRResNet method)
(mmedit.models.backbones.PConvEncoderDecoder method)
(mmedit.models.backbones.PlainDecoder method)
(mmedit.models.backbones.ResNetDec method)
(mmedit.models.backbones.ResnetGenerator method)
(mmedit.models.backbones.RRDBNet method)
(mmedit.models.backbones.SRCNN method)
(mmedit.models.backbones.TOFlow method)
(mmedit.models.backbones.UnetGenerator method)
(mmedit.models.BaseMattor method)
(mmedit.models.BaseModel method)
(mmedit.models.BasicRestorer method)
(mmedit.models.common.PixelShufflePack method)
(mmedit.models.common.ResidualBlockNoBN method)
(mmedit.models.components.DeepFillv1Discriminators method)
(mmedit.models.components.GLDiscs method)
(mmedit.models.components.ModifiedVGG method)
(mmedit.models.components.MultiLayerDiscriminator method)
(mmedit.models.components.PatchDiscriminator method)
(mmedit.models.CycleGAN method)
(mmedit.models.losses.PerceptualVGG method)
(mmedit.models.OneStageInpaintor method)
(mmedit.models.Pix2Pix method)
(mmedit.models.SRGAN method)
L
L1CompositionLoss (class in mmedit.models.losses)
L1Evaluation (class in mmedit.core)
L1Loss (class in mmedit.models.losses)
LinearLrUpdaterHook (class in mmedit.core)
LinearModule (class in mmedit.models.common)
load_annotations() (mmedit.datasets.AdobeComp1kDataset method)
(mmedit.datasets.BaseDataset method)
(mmedit.datasets.GenerationPairedDataset method)
(mmedit.datasets.GenerationUnpairedDataset method)
(mmedit.datasets.ImgInpaintingDataset method)
(mmedit.datasets.SRAnnotationDataset method)
(mmedit.datasets.SRFolderDataset method)
(mmedit.datasets.SRLmdbDataset method)
(mmedit.datasets.SRREDSDataset method)
(mmedit.datasets.SRVid4Dataset method)
(mmedit.datasets.SRVimeo90KDataset method)
LoadImageFromFile (class in mmedit.datasets.pipelines)
LoadImageFromFileList (class in mmedit.datasets.pipelines)
LoadMask (class in mmedit.datasets.pipelines)
LoadPairedImageFromFile (class in mmedit.datasets.pipelines)
M
make_layer() (in module mmedit.models.common)
mask_correlation_map() (mmedit.models.common.ContextualAttentionModule method)
mask_reduce_loss() (in module mmedit.models.losses)
MaskConvModule (class in mmedit.models.common)
MaskedTVLoss (class in mmedit.models.losses)
MergeFgAndBg (class in mmedit.datasets.pipelines)
mmedit.core
module
mmedit.datasets
module
mmedit.datasets.pipelines
module
mmedit.models
module
mmedit.models.backbones
module
mmedit.models.common
module
mmedit.models.components
module
mmedit.models.losses
module
mmedit.utils
module
ModCrop (class in mmedit.datasets.pipelines)
ModifiedVGG (class in mmedit.models.components)
module
mmedit.core
mmedit.datasets
mmedit.datasets.pipelines
mmedit.models
mmedit.models.backbones
mmedit.models.common
mmedit.models.components
mmedit.models.losses
mmedit.utils
MSECompositionLoss (class in mmedit.models.losses)
MSELoss (class in mmedit.models.losses)
MSRResNet (class in mmedit.models.backbones)
MultiLayerDiscriminator (class in mmedit.models.components)
N
Normalize (class in mmedit.datasets.pipelines)
normalize() (mmedit.models.backbones.TOFlow method)
O
OneStageInpaintor (class in mmedit.models)
P
Pad (class in mmedit.datasets.pipelines)
PairedRandomCrop (class in mmedit.datasets.pipelines)
parse_losses() (mmedit.models.BaseModel method)
PartialConv2d (class in mmedit.models.common)
patch_copy_deconv() (mmedit.models.common.ContextualAttentionModule method)
patch_correlation() (mmedit.models.common.ContextualAttentionModule method)
PatchDiscriminator (class in mmedit.models.components)
PConvDecoder (class in mmedit.models.backbones)
PConvEncoder (class in mmedit.models.backbones)
PConvEncoderDecoder (class in mmedit.models.backbones)
PConvInpaintor (class in mmedit.models)
PerceptualLoss (class in mmedit.models.losses)
PerceptualVGG (class in mmedit.models.losses)
PerturbBg (class in mmedit.datasets.pipelines)
Pix2Pix (class in mmedit.models)
PixelShufflePack (class in mmedit.models.common)
PlainDecoder (class in mmedit.models.backbones)
PlainRefiner (class in mmedit.models.components)
prepare_test_data() (mmedit.datasets.BaseDataset method)
(mmedit.datasets.GenerationUnpairedDataset method)
prepare_train_data() (mmedit.datasets.BaseDataset method)
(mmedit.datasets.GenerationUnpairedDataset method)
process_unknown_mask() (mmedit.models.common.GCAModule method)
propagate_alpha_feature() (mmedit.models.common.GCAModule method)
psnr() (in module mmedit.core)
Q
query() (mmedit.models.common.GANImageBuffer method)
R
RandomAffine (class in mmedit.datasets.pipelines)
RandomJitter (class in mmedit.datasets.pipelines)
RandomLoadResizeBg (class in mmedit.datasets.pipelines)
RandomMaskDilation (class in mmedit.datasets.pipelines)
RandomTransposeHW (class in mmedit.datasets.pipelines)
reduce_loss() (in module mmedit.models.losses)
reorder_image() (in module mmedit.core)
RepeatDataset (class in mmedit.datasets)
RescaleToZeroOne (class in mmedit.datasets.pipelines)
ResGCADecoder (class in mmedit.models.backbones)
ResGCAEncoder (class in mmedit.models.backbones)
ResidualBlockNoBN (class in mmedit.models.common)
ResidualBlockWithDropout (class in mmedit.models.common)
Resize (class in mmedit.datasets.pipelines)
ResNetDec (class in mmedit.models.backbones)
ResNetEnc (class in mmedit.models.backbones)
ResnetGenerator (class in mmedit.models.backbones)
ResShortcutDec (class in mmedit.models.backbones)
ResShortcutEnc (class in mmedit.models.backbones)
restore_shape() (mmedit.models.BaseMattor method)
RRDBNet (class in mmedit.models.backbones)
S
save_image() (mmedit.models.BaseMattor method)
save_visualization() (mmedit.models.OneStageInpaintor method)
(mmedit.models.TwoStageInpaintor method)
scale_bbox() (in module mmedit.models.common)
scan_folder() (mmedit.datasets.BaseGenerationDataset static method)
(mmedit.datasets.BaseSRDataset static method)
set_requires_grad() (in module mmedit.models.common)
setup() (mmedit.models.CycleGAN method)
(mmedit.models.Pix2Pix method)
SimpleEncoderDecoder (class in mmedit.models.backbones)
SimpleGatedConvModule (class in mmedit.models.common)
spatial_discount_mask() (mmedit.datasets.pipelines.GetSpatialDiscountMask method)
SRAnnotationDataset (class in mmedit.datasets)
SRCNN (class in mmedit.models.backbones)
SRFolderDataset (class in mmedit.datasets)
SRGAN (class in mmedit.models)
SRLmdbDataset (class in mmedit.datasets)
SRREDSDataset (class in mmedit.datasets)
SRVid4Dataset (class in mmedit.datasets)
SRVimeo90KDataset (class in mmedit.datasets)
ssim() (in module mmedit.core)
T
TemporalReverse (class in mmedit.datasets.pipelines)
tensor2img() (in module mmedit.core)
TOFlow (class in mmedit.models.backbones)
ToTensor (class in mmedit.datasets.pipelines)
train() (mmedit.models.backbones.PConvEncoder method)
train_step() (mmedit.models.BaseMattor method)
(mmedit.models.BaseModel method)
(mmedit.models.BasicRestorer method)
(mmedit.models.CycleGAN method)
(mmedit.models.DeepFillv1Inpaintor method)
(mmedit.models.ESRGAN method)
(mmedit.models.GLInpaintor method)
(mmedit.models.OneStageInpaintor method)
(mmedit.models.PConvInpaintor method)
(mmedit.models.Pix2Pix method)
(mmedit.models.SRGAN method)
(mmedit.models.TwoStageInpaintor method)
two_stage_loss() (mmedit.models.DeepFillv1Inpaintor method)
(mmedit.models.TwoStageInpaintor method)
TwoStageInpaintor (class in mmedit.models)
U
UnetGenerator (class in mmedit.models.backbones)
UnetSkipConnectionBlock (class in mmedit.models.common)
V
val_step() (mmedit.models.BaseModel method)
(mmedit.models.BasicRestorer method)
(mmedit.models.CycleGAN method)
(mmedit.models.OneStageInpaintor method)
(mmedit.models.Pix2Pix method)
VGG16 (class in mmedit.models.backbones)
VisualizationHook (class in mmedit.core)
W
with_refiner() (mmedit.models.BaseMattor property)
Read the Docs
v: latest
Versions
latest
stable
Downloads
On Read the Docs
Project Home
Builds