A Jittor implementation of Conditional GAN (CGAN)

Dataset

• MNIST can download by jittor

• Fashion-MNIST jittor doesn’t provide. You can download by pytorch.

How to use

• train

  python cgan_jittor.py

  every epoch, it will save a image in img_jittor_xxx folder.

  the model will be saved as d_xxx.pkl and g_xxx.pkl.

• generate image

  python gen_image.py

  it will generate images in fid_xxx folder.

evaluate

use FID score to evaluate.

provide pytorch implementation.

python fid_score.py