Shortcuts

Source code for mmedit.models.editors.aotgan.aot_encoder_decoder

# Copyright (c) OpenMMLab. All rights reserved.

from mmedit.registry import MODELS
from ..global_local import GLEncoderDecoder


@MODELS.register_module()
[docs]class AOTEncoderDecoder(GLEncoderDecoder): """Encoder-Decoder used in AOT-GAN model. This implementation follows: Aggregated Contextual Transformations for High-Resolution Image Inpainting The architecture of the encoder-decoder is: (conv2d x 3) --> (dilated conv2d x 8) --> (conv2d or deconv2d x 3). Args: encoder (dict): Config dict to encoder. decoder (dict): Config dict to build decoder. dilation_neck (dict): Config dict to build dilation neck. """ def __init__(self, encoder=dict(type='AOTEncoder'), decoder=dict(type='AOTDecoder'), dilation_neck=dict(type='AOTBlockNeck')): super().__init__() self.encoder = MODELS.build(encoder) self.decoder = MODELS.build(decoder) self.dilation_neck = MODELS.build(dilation_neck)
Read the Docs v: latest
Versions
master
latest
stable
zyh-re-docs
zyh-doc-notfound-extend
zyh-api-rendering
Downloads
pdf
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.