Code for the Paper "Diffusion Models for Handwriting Generation"

Last update: Dec 21, 2022

Related tags

Deep Learning Diffusion-Handwriting-Generation

Overview

Code for the Paper "Diffusion Models for Handwriting Generation": https://arxiv.org/abs/2011.06704

Project written in python 3.7, Tensorflow 2.3

To run the model, install required packages with pip install -r requirements.txt

Then run inference.py and specify arguments as needed

To retrain model, run train.py, and specify arguments to change hyperparameters All models will be saved in the ./weights directory

Before running the training script, download the following things from https://fki.tic.heia-fr.ch/databases/download-the-iam-on-line-handwriting-database

data/lineStrokes-all.tar.gz - the stroke xml for the online dataset data/lineImages-all.tar.gz - the images for the offline dataset ascii-all.tar.gz - the text labels for the dataset extract these contents and put them in the ./data directory

You might also like...

PyTorch code for the paper: FeatMatch: Feature-Based Augmentation for Semi-Supervised Learning

FeatMatch: Feature-Based Augmentation for Semi-Supervised Learning This is the PyTorch implementation of our paper: FeatMatch: Feature-Based Augmentat

43 Nov 19, 2022

Code for the paper A Theoretical Analysis of the Repetition Problem in Text Generation

A Theoretical Analysis of the Repetition Problem in Text Generation This repository share the code for the paper "A Theoretical Analysis of the Repeti

37 Nov 21, 2022

Code for our ICASSP 2021 paper: SA-Net: Shuffle Attention for Deep Convolutional Neural Networks

SA-Net: Shuffle Attention for Deep Convolutional Neural Networks (paper) By Qing-Long Zhang and Yu-Bin Yang [State Key Laboratory for Novel Software T

199 Jan 8, 2023

Open source repository for the code accompanying the paper 'Non-Rigid Neural Radiance Fields Reconstruction and Novel View Synthesis of a Deforming Scene from Monocular Video'.

Non-Rigid Neural Radiance Fields This is the official repository for the project "Non-Rigid Neural Radiance Fields: Reconstruction and Novel View Synt

296 Dec 29, 2022

Code for the Shortformer model, from the paper by Ofir Press, Noah A. Smith and Mike Lewis.

Shortformer This repository contains the code and the final checkpoint of the Shortformer model. This file explains how to run our experiments on the

138 Apr 15, 2022

PyTorch code for ICLR 2021 paper Unbiased Teacher for Semi-Supervised Object Detection

Unbiased Teacher for Semi-Supervised Object Detection This is the PyTorch implementation of our paper: Unbiased Teacher for Semi-Supervised Object Detection

366 Dec 28, 2022

Official code for paper "Optimization for Oriented Object Detection via Representation Invariance Loss".

Optimization for Oriented Object Detection via Representation Invariance Loss By Qi Ming, Zhiqiang Zhou, Lingjuan Miao, Xue Yang, and Yunpeng Dong. Th

56 Nov 28, 2022

Code for our CVPR 2021 paper "MetaCam+DSCE"

Joint Noise-Tolerant Learning and Meta Camera Shift Adaptation for Unsupervised Person Re-Identification (CVPR'21) Introduction Code for our CVPR 2021

59 Oct 31, 2022

Code for our CVPR2021 paper coordinate attention

Coordinate Attention for Efficient Mobile Network Design (preprint) This repository is a PyTorch implementation of our coordinate attention (will appe

726 Jan 5, 2023

Comments

StyleExtractor input shape
In StyleExtractor class, the MobileNetV2 model receives an input shape of (96, 96, 3). However, writer image is rarely a square, which leads to the problem of incompatible input shape for StyleExtractor. Here's one example error when running inference with default setting:

Input 0 of layer "mobilenetv2_1.00_96" is incompatible with the layer: expected shape=(None, 96, 96, 3), found shape=(96, 96, 1400, 3) Call arguments received by layer "style_extractor" "f"(type StyleExtractor): im=tf.Tensor(shape=(96, 96, 1400, 1), dtype=uint8) im2=None get_similarity=False training=False

How do we deal with the incompatible shape here? Is it safe to assume that reshaping the writer image to a square doesn't affect the model's performance?
opened by longxvu 2
How to get file "train_strokes.p"

https://github.com/tcl9876/Diffusion-Handwriting-Generation/blob/26a2dc33648b26b49eb1d11af118e0dc71fa8698/train.py#L88 In this line, what is the file "train_strokes.p" and how to get it? thanks a lot.

opened by XDUWQ 2
Issue when Inference in Batch

Hi authors, When I try to inference in individual text, it works well. But when I try to infere in batch, I give input as a list, just like this ["abcdefu", "1234567", "9876543"]. There is a bug happened, it called "Incompatible shapes: [1,45,384] vs. [3,384] [Op:Mul]"

How can I fix this?

opened by suonbo 1

Owner

GitHub

This is the official source code for SLATE. We provide the code for the model, the training code, and a dataset loader for the 3D Shapes dataset. This code is implemented in Pytorch.

SLATE This is the official source code for SLATE. We provide the code for the model, the training code and a dataset loader for the 3D Shapes dataset.

66 Dec 26, 2022

Code for paper ECCV 2020 paper: Who Left the Dogs Out? 3D Animal Reconstruction with Expectation Maximization in the Loop.

Who Left the Dogs Out? Evaluation and demo code for our ECCV 2020 paper: Who Left the Dogs Out? 3D Animal Reconstruction with Expectation Maximization

29 Dec 28, 2022

TensorFlow code for the neural network presented in the paper: "Structural Language Models of Code" (ICML'2020)

SLM: Structural Language Models of Code This is an official implementation of the model described in: "Structural Language Models of Code" [PDF] To ap

73 Nov 6, 2022

Code for the prototype tool in our paper "CoProtector: Protect Open-Source Code against Unauthorized Training Usage with Data Poisoning".

CoProtector Code for the prototype tool in our paper "CoProtector: Protect Open-Source Code against Unauthorized Training Usage with Data Poisoning".

1 Oct 26, 2021

Code to use Augmented Shapiro Wilks Stopping, as well as code for the paper "Statistically Signifigant Stopping of Neural Network Training"

This codebase is being actively maintained, please create and issue if you have issues using it Basics All data files are included under losses and ea

32 Nov 9, 2021