Generative Handwriting using LSTM Mixture Density Network with TensorFlow

Last update: Nov 24, 2022

Related tags

Deep Learning write-rnn-tensorflow

Overview

Generative Handwriting Demo using TensorFlow

An attempt to implement the random handwriting generation portion of Alex Graves' paper.

See my blog post at blog.otoro.net for more information.

How to use

I tested the implementation on TensorFlow r0.11 and Pyton 3. I also used the following libraries to help:

svgwrite
IPython.display.SVG
IPython.display.display
xml.etree.ElementTree
argparse
pickle

Training

You will need permission from these wonderful people people to get the IAM On-Line Handwriting data. Unzip lineStrokes-all.tar.gz into the data subdirectory, so that you end up with data/lineStrokes/a01, data/lineStrokes/a02, etc. Afterwards, running python train.py will start the training process.

A number of flags can be set for training if you wish to experiment with the parameters. The default values are in train.py

--rnn_size RNN_SIZE             size of RNN hidden state
--num_layers NUM_LAYERS         number of layers in the RNN
--model MODEL                   rnn, gru, or lstm
--batch_size BATCH_SIZE         minibatch size
--seq_length SEQ_LENGTH         RNN sequence length
--num_epochs NUM_EPOCHS         number of epochs
--save_every SAVE_EVERY         save frequency
--grad_clip GRAD_CLIP           clip gradients at this value
--learning_rate LEARNING_RATE   learning rate
--decay_rate DECAY_RATE         decay rate for rmsprop
--num_mixture NUM_MIXTURE       number of gaussian mixtures
--data_scale DATA_SCALE         factor to scale raw data down by
--keep_prob KEEP_PROB           dropout keep probability

Generating a Handwriting Sample

I've included a pretrained model in /save so it should work out of the box. Running python sample.py --filename example_name --sample_length 1000 will generate 4 .svg files for each example, with 1000 points.

IPython interactive session.

If you wish to experiment with this code interactively, just run %run -i sample.py in an IPython console, and then the following code is an example on how to generate samples and show them inside IPython.

[strokes, params] = model.sample(sess, 800)
draw_strokes(strokes, factor=8, svg_filename = 'sample.normal.svg')
draw_strokes_random_color(strokes, factor=8, svg_filename = 'sample.color.svg')
draw_strokes_random_color(strokes, factor=8, per_stroke_mode = False, svg_filename = 'sample.multi_color.svg')
draw_strokes_eos_weighted(strokes, params, factor=8, svg_filename = 'sample.eos.svg')
draw_strokes_pdf(strokes, params, factor=8, svg_filename = 'sample.pdf.svg')

Have fun-

License

MIT

Generative Handwriting using LSTM Mixture Density Network with TensorFlow

Related tags

Overview

Generative Handwriting Demo using TensorFlow

How to use

Training

Generating a Handwriting Sample

IPython interactive session.

License

Owner

hardmaru

A PyTorch implementation of Radio Transformer Networks from the paper "An Introduction to Deep Learning for the Physical Layer".

The Python code for the paper A Hybrid Quantum-Classical Algorithm for Robust Fitting

A real-time approach for mapping all human pixels of 2D RGB images to a 3D surface-based model of the body

Self-Supervised Learning

This repo provides code for QB-Norm (Cross Modal Retrieval with Querybank Normalisation)

Self-Supervised Multi-Frame Monocular Scene Flow (CVPR 2021)

The Multi-Mission Maximum Likelihood framework (3ML)

Official implementation of "Towards Good Practices for Efficiently Annotating Large-Scale Image Classification Datasets" (CVPR2021)

This repo contains the code required to train the multivariate time-series Transformer.

Implementation of EMNLP 2017 Paper "Natural Language Does Not Emerge 'Naturally' in Multi-Agent Dialog" using PyTorch and ParlAI

Large-scale language modeling tutorials with PyTorch

Self-Supervised Document-to-Document Similarity Ranking via Contextualized Language Models and Hierarchical Inference

competitions-v2

Apply a perspective transformation to a raster image inside Inkscape (no need to use an external software such as GIMP or Krita).

Leveraging OpenAI's Codex to solve cornerstone problems in Music

Auto-updating data to assist in investment to NEPSE

[CVPR21] LightTrack: Finding Lightweight Neural Network for Object Tracking via One-Shot Architecture Search

Introduction to Statistics and Basics of Mathematics for Data Science - The Hacker's Way

A Python module for the generation and training of an entry-level feedforward neural network.

Using VideoBERT to tackle video prediction