Awesome Open Source
Awesome Open Source


PyTorch Implementation of SUM-GAN

Changes from Original paper

  • Video feature extractor
    • GoogleNet pool5 (1024) => ResNet-101 pool5 (2048)
    • Followed by linear projection to 500-dim
  • Stable GAN Training
    • Discriminator's learning rate: 1e-5 (Others: 1e-4)
    • Fix Discriminators' parameters for first 15 steps at every epoch.

Model figures

Model figure 1

Model figure 2

Model figure 3


Algorithm figure

Get A Weekly Email With Trending Projects For These Topics
No Spam. Unsubscribe easily at any time.
python (53,199
pytorch (2,319
video (839
gan (344
vae (69
summarization (40