https://github.com/iver56/image-regression
A DNN that takes in x, y and outputs colors at that point to approximately "paint" an image. A toy problem.
https://github.com/iver56/image-regression
deep-neural-networks dnn image keras python regression
Last synced: about 1 month ago
JSON representation
A DNN that takes in x, y and outputs colors at that point to approximately "paint" an image. A toy problem.
- Host: GitHub
- URL: https://github.com/iver56/image-regression
- Owner: iver56
- Created: 2017-02-17T19:04:02.000Z (about 8 years ago)
- Default Branch: master
- Last Pushed: 2022-07-21T13:54:34.000Z (almost 3 years ago)
- Last Synced: 2025-03-17T14:52:50.019Z (about 2 months ago)
- Topics: deep-neural-networks, dnn, image, keras, python, regression
- Language: Python
- Homepage:
- Size: 1.58 MB
- Stars: 47
- Watchers: 4
- Forks: 14
- Open Issues: 6
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# Image regression
A toy application that learns a mapping from (x, y) coordinates to color. Uses pytorch and pytorch-lightning.
## Setup
Set up your python environment with conda like this:
`conda env create`
## Usage
### Train on a single greyscale image
`python train.py -i keyboard.png --num-epochs 150`
### Train on a single color image (RGBA)
TODO
### Train on multiple images
`python train.py -i boxy_stripes2.png boxy_stripes2_30.png boxy_stripes2_60.png boxy_stripes2_90.png`
### Train on multiple color images (RGBA)
TODO
### Interpolate between the images:
`python interpolate_between_many.py --width 64 --height 64 --num-images 4 --model boxy_stripes2_boxy_stripes2_30_boxy_stripes2_60_bo_2f59b848.onnx --use-cuda 0`
## Examples
| Name | Original | Learned image |
| ---- | -------- | ------------- |
| Keyboard |  |  |
| 8x8 Checkerboard |  |  |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.
