Skip to content

PyTorch implementation of DRAW: A Recurrent Neural Network For Image Generation trained on Devanagari dataset.

Notifications You must be signed in to change notification settings

Natsu6767/Generating-Devanagari-Using-DRAW

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

28 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Generating-Devanagari-Using-DRAW

PyTorch implementation of DRAW: A Recurrent Neural Network For Image Generation on the task of generating Devanagari Characters.

Generated Data Animation

Training

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

Training Loss Curves

Generating New Images

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.

Results

Devanagari Training Data Generated Devanagari After 50 Epochs
Devanagari Numbers Only Training Data Generated Devanagari Numbers After 50 Epochs

Some more generated images:

References

  1. Karol Gregor, Ivo Danihelka, Alex Graves, Danilo Jimenez Rezende, Daan Wierstra. DRAW: A Recurrent Neural Network For Image Generation. [arxiv]
  2. ericjang/draw [repo]
  3. What is DRAW (Deep Recurrent Attentive Writer)? [blog]

Data

The Devanagari Character dataset is available on kaggle. (Source)

About

PyTorch implementation of DRAW: A Recurrent Neural Network For Image Generation trained on Devanagari dataset.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages