triztex / handwriting-generation Goto Github PK

Download the checkpoint content from this link Put the contents of the above downloaded folder into checkpoints folder. The software requirements are listed in the requirements.txt file.

Create a 'logs' folder before running, files are saved as usage_demo in img folder. Further instructions to run are in run.py

python run.py

A pretrained model is included, for training your own model:

In order to train a model, data must be downloaded and placed in this directory.

Follow the download instructions here http://www.fki.inf.unibe.ch/databases/iam-on-line-handwriting-database.

Only a subset of the downloaded data is required. Move the relevant download data so the directory structure is as folllows:

data/
├── raw/
│   ├── ascii/
│   ├── lineStrokes/
│   ├── original/
|   blacklist.npy

Once this is completed, run prepare_data.py extract the data and dump it to numpy files.

To train the model, run rnn.py. This takes a couple days on a single Tesla K80.

Made with ❤️ by DS Community SRM