akanimax/attn_gan_pytorch
python package for self-attention gan implemented as extension of PyTorch nn.Module. paper -> https://arxiv.org/abs/1805.08318
This is a Python package for PyTorch developers working with generative adversarial networks (GANs). It provides an implementation of the self-attention mechanism, allowing you to incorporate advanced image-based attention into your neural network models. Developers can use this to enhance the quality of generated images or improve feature learning in image processing tasks.
No commits in the last 6 months. Available on PyPI.
Use this if you are a PyTorch developer building or experimenting with GANs and want to integrate self-attention or other generic image-based attention layers into your models.
Not ideal if you are looking for a ready-to-use application for generating images without writing Python code, or if you are not familiar with PyTorch and neural network development.
Stars
19
Forks
5
Language
Python
License
MIT
Category
Last pushed
Sep 14, 2018
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/akanimax/attn_gan_pytorch"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
huggingface/pytorch-pretrained-BigGAN
🦋A PyTorch implementation of BigGAN with pretrained weights and conversion scripts.
torchgan/torchgan
Research Framework for easy and efficient training of GANs based on Pytorch
metal3d/keras-video-generators
Keras generators to generate sequences from videos as input
GANs-in-Action/gans-in-action
Companion repository to GANs in Action: Deep learning with Generative Adversarial Networks
junyanz/pytorch-CycleGAN-and-pix2pix
Image-to-Image Translation in PyTorch