|
--- |
|
license: mit |
|
library_name: pytorch |
|
tags: |
|
- gan |
|
- sngan |
|
- huggan |
|
dataset: ChainYo/rvl-cdip-invoice |
|
--- |
|
|
|
## Model description |
|
|
|
SN-GAN implementation with PyTorch-Lightning to generate Documents. |
|
|
|
## Generated samples |
|
|
|
<img src="https://github.com/ChainYo/docugan/blob/master/documents_samples.png" width="400" height="1200"> |
|
|
|
Project repository: [DocuGAN](https://github.com/ChainYo/docugan). |
|
|
|
## Usage |
|
|
|
You see the genration tool on HuggingFace by trying the [space demo](https://huggingface.co/spaces/ChainYo/DocuGAN). |
|
|
|
## Training data |
|
|
|
For training, I used the invoices subpart of `RVL-CDIP` dataset. Find the full dataset [here](https://huggingface.co/datasets/ChainYo/rvl-cdip) |
|
|
