|
--- |
|
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/c4d9ae7bd5a63d4aedee40e7a2670222170e44c3/documents_samples.png" width="400" height="600"> |
|
|
|
Project repository: [DocuGAN](https://github.com/ChainYo/docugan). |
|
|
|
## Usage |
|
|
|
You can see the tool to generate document 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) |
|
|
