如何在colab中导入AdversarialModel、simple_gan、gan_targets

Question

我正在导入keras_adversarial并克隆它的git，但它不能导入'AdversarialModel‘

我在gans上工作，构建生成器和鉴别器。现在我想将它们组合在一起，但是colab在导入keras_adversarial的模块时出现错误

import keras_adversarial
from keras_adversarial import AdversarialModel, simple_gan, gan_targets
from keras_adversarial import AdversarialOptimizerSimultaneous, 
normal_latent_sampling

1 import keras_adversarial
----> 2 from keras_adversarial import AdversarialModel, simple_gan, 
gan_targets
3 from keras_adversarial import AdversarialOptimizerSimultaneous, 
normal_latent_sampling

ImportError: cannot import name 'AdversarialModel'

Answer 1

Colab环境只安装了一组核心软件包。要添加一些第三方包作为keras_adversarial，您可以直接在Colab cell中执行安装脚本，在每个命令前使用!符号，以指示这些是命令行bash代码，而不是Python。

在您的情况下，您需要执行以下操作：

!git clone https://github.com/bstriner/keras_adversarial.git
!cd keras_adversarial
!python setup.py install