是否有使用MxNet制作卷积自动编码器的玩具例子？

Question

我正在寻找使用MxNet实现卷积自动编码器。但是基于全连接网络的自动编码器只有一个例子，那就是这里。在github中也有一个问题提出了类似的问题，但得到的答复很少。是否有使用MxNet实现卷积自动编码器的玩具示例？

Answer 1

请在Mxnet胶子中找到Conv自动编码器模型的示例。引用自这里的代码。在胶子中以标准的方式训练这个模型。

from mxnet import gluon as g

class CNNAutoencoder(g.nn.HybridBlock):
    def __init__(self):
        super(CNNAutoencoder, self).__init__()
        with self.name_scope():
            self.encoder = g.nn.HybridSequential('encoder_')
            with self.encoder.name_scope():
                self.encoder.add(g.nn.Conv2D(16, 3, strides=3, padding=1, activation='relu'))
                self.encoder.add(g.nn.MaxPool2D(2, 2))
                self.encoder.add(g.nn.Conv2D(8, 3, strides=2, padding=1, activation='relu'))
                self.encoder.add(g.nn.MaxPool2D(2, 1))

            self.decoder = g.nn.HybridSequential('decoder_')
            with self.decoder.name_scope():
                self.decoder.add(g.nn.Conv2DTranspose(16, 3, strides=2, activation='relu'))
                self.decoder.add(g.nn.Conv2DTranspose(8, 5, strides=3, padding=1, activation='relu'))
                self.decoder.add(g.nn.Conv2DTranspose(1, 2, strides=2, padding=1, activation='tanh'))

    def forward(self, x):
        x = self.encoder(x)
        x = self.decoder(x)
        return x

model = CNNAutoencoder()
model.hybridize()

Answer 2

在mxnet中仍然没有卷积自动编码器的例子，尽管在这方面有一些研究进展。无论如何，在一张票 github中有MxNet，但是它仍然是开放的。我们非常欢迎您做出贡献，例如，通过从Keras迁移代码。