bendangnuksung / Image-OutPainting

πŸ– Keras Implementation of Painting outside the box

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Keras implementation of Image OutPainting

This is an implementation of Painting Outside the Box: Image Outpainting paper from Standford University. Some changes have been made to work with 256*256 image:

  • Added Identity loss i.e from generated image to the original image
  • Removed patches from training data. (training pipeline)
  • Replaced masking with cropping. (training pipeline)
  • Added convolution layers.

Results

The model was train with 3500 scrapped beach data with agumentation totalling upto 10500 images for 25 epochs. Demo

Recursive painting

Demo

Install Requirements

sudo apt-get install curl
sudo pip3 install -r requirements.txt

Get Started

  1. Prepare Data:
    # Downloads the beach data and converts to numpy batch data
    # saves the Numpy batch data to 'data/prepared_data/'
    sh prepare_data.sh
  2. Build Model
    • To build Model from scratch you can directly run 'outpaint.ipynb'
      OR
    • You can Download my trained model and move it to 'checkpoint/' and run it.

References

About

πŸ– Keras Implementation of Painting outside the box


Languages

Language:Jupyter Notebook 77.2%Language:Python 20.5%Language:Shell 2.2%