| --- |
| license: mit |
| library_name: pytorch |
| tags: |
| - gan |
| - sngan |
| - huggan |
| - unconditional-image-generation |
| datasets: |
| - ChainYo/rvl-cdip-invoice |
| --- |
| |
| ## Model description |
|
|
| SN-GAN implementation with PyTorch-Lightning to generate Documents. |
|
|
| ## Generated samples |
|
|
| <img src="https://raw.githubusercontent.com/ChainYo/docugan/master/documents_samples.png" width="400" height="1200"> |
|
|
| 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) |
|
|
