🚀 Want to run this model with an API? Get started

cjwbw/instruct-pix2pix

Public
Edit images with human instructions
5.5K runs

Run time and cost

Predictions run on Nvidia A100 GPU hardware. Predictions typically complete within 10 seconds.

InstructPix2Pix: Learning to Follow Image Editing Instructions

Edit images with written instructions:

Abstract

We propose a method for editing images from human instructions: given an input image and a written instruction that tells the model what to do, our model follows these instructions to edit the image. To obtain training data for this problem, we combine the knowledge of two large pretrained models---a language model (GPT-3) and a text-to-image model (Stable Diffusion)---to generate a large dataset of image editing examples. Our conditional diffusion model, InstructPix2Pix, is trained on our generated data, and generalizes to real images and user-written instructions at inference time. Since it performs edits in the forward pass and does not require per-example fine-tuning or inversion, our model edits images quickly, in a matter of seconds. We show compelling editing results for a diverse collection of input images and written instructions.

See the following for further details: Project Page | Paper | Data

Improving your outputs:

If you're not getting the quality of output you'd like, see here for some tips.

BibTeX

@article{brooks2022instructpix2pix,
  title={InstructPix2Pix: Learning to Follow Image Editing Instructions},
  author={Brooks, Tim and Holynski, Aleksander and Efros, Alexei A},
  journal={arXiv preprint arXiv:2211.09800},
  year={2022}
}