PyTorch implementation of DRAW: A Recurrent Neural Network For Image Generation trained on Devanagari dataset.
PyTorch implementation of DRAW: A Recurrent Neural Network For Image Generation on the task of generating Devanagari Characters.
Download the data and place it in the data/ directory. Run train.py
to start training. To change the hyperparameters of the network, update the values in the param
dictionary in train.py
.
Loss Curve
To generate new images run generate.py
.
python3 evaluate.py -load_path /path/to/pth/checkpoint -num_output n
The checkpoint file for the model trained for 50 epochs is present in checkpoint/ directory.
Devanagari Training Data | Generated Devanagari After 50 Epochs |
---|---|
Devanagari Numbers Only Training Data | Generated Devanagari Numbers After 50 Epochs |
---|---|
The Devanagari Character dataset is available on kaggle. (Source)