Skip to content

Towards Diverse and Natural Image Descriptions via a Conditional GAN

License

Notifications You must be signed in to change notification settings

doubledaibo/gancaption_iccv2017

Repository files navigation

Based on Neuraltalk2 and SeqGAN. Special thanks to the authors!

Testing codes and related codes will be added gradually.

Brief Explanation

  • Without images as conditions

    • utils/DataLoaderSeq.lua
    • utils/netUtilsSeq.lua
    • utils/netTrainSeq.lua
    • dSeq.lua
    • gSeq.lua
    • trainGANSeqPolicy.lua
    • Can be used to validate the codes for rollouts. If there are bugs, g will eventually generate nonsense
  • With images as conditions, without finetuning the image encoder

    • utils/DataLoader.lua
    • utils/netUtils.lua
    • utils/netTrain.lua
    • dDotprod.lua
    • gNoise.lua
    • trainGANPolicyFeatNoise.lua

About

Towards Diverse and Natural Image Descriptions via a Conditional GAN

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages