使用哪种深度学习框架生成预测图像

Question

假设您有一个相对相似图像的数据集，对于每个图像，图像的前半部分是模型的输入，后半部分是模型的输出。换句话说，您想教模型单个图像的前半部分，并让其生成后半部分。哪个框架最适合此问题？

Answer 1

您可以阅读有关生成对抗网络（GAN）的信息。然后使用任一PyTorch 或Python上的Tensorflow，尽管PyTorch以GAN闻名。

Answer 2

Pytorch深度学习框架应该为您工作，因为它具有许多研究社区的支持，并且在研究阶段，许多深度学习算法都是在Pytorch中实现的。

关于提到的任务，您应该尝试使用GAN并在下面提到一些特定的GANS算法：

• DCGAN from official pytorch repo.
• DCGAN Unofficial implementation