Skip to content

A DNN that takes in x, y and outputs colors at that point to approximately "paint" an image. A toy problem.

Notifications You must be signed in to change notification settings

gep/image-regression

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

28 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Image regression

A toy application that learns a mapping from (x, y) coordinates to color.

Usage

Train on a single greyscale image

python train.py -i keyboard.png --num-epochs 1000

Train on a single color image (RGBA)

python train.py -i landscape.png --num-epochs 1000

Train on multiple images

python train_many.py -i boxy_stripes2.png boxy_stripes2_30.png boxy_stripes2_60.png boxy_stripes2_90.png

Then interpolate between the images:

python interpolate_between_many.py --num-images 4 --model output/that_model_filename.h5

Examples

Name Original Learned image
Keyboard Original image Learned image
8x8 Checkerboard Original image Learned image

The following animation visualizes the output of a neural network that was trained on 12 different images (different rotations of boxy stripes). The input vectors are constructed to interpolate between the 12 images, so we get a kind of morphing effect.

Boxy stripes

About

A DNN that takes in x, y and outputs colors at that point to approximately "paint" an image. A toy problem.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%