Skip to content

WisconsinAIVision/edit-one-for-all

Repository files navigation

✏️ Edit One for All: Interactive Batch Image Editing

arXiv - project page

Given an edit specified by users in an example image (e.g., dog pose), Our method can automatically transfer that edit to other test images (e.g., all dog same pose).

✏️ Edit One for All: Interactive Batch Image Editing (CVPR 2024)
By Thao Nguyen, Utkarsh Ojha, Yuheng Li, Haotian Liu, Yong Jae Lee
🦡 University of Wisconsin-Madison


Getting Started

This repo is heavily built upon DragGAN. Please refer to the original repo for more details about installation/download checkpoints.

#- clone this repo
git clone https://github.com/WisconsinAIVision/edit-one-for-all.git
cd edit-one-for-all

#- install packes
conda env create -f environment.yml
conda activate stylegan3
pip install -r requirements.txt

#- download checkpoints
python scripts/download_model.py

If you want to try other pretrained model, put them under ./checkpoints/ folder.

Usage

Launch the gradio demo by:

python my_gradio.py --port 7681

Then, open your browser and go to http://localhost:7681/ to interact with the demo.

Galleries

As users adjust the editing strength in the example image (top row), all test images will be automatically updated. (Red bounding boxes indicate the edit according to the drag points).

  • Interactive Dog Pose:

  • Interactive Anime Hair Length:

  • Interactive Mountain Height:

  • Interactive Human Pose:

  • Interactive Face Slimming:

  • Interactive Tiger Roar:

Related Works

Special thanks to DragGAN for making the code available!

BibTeX

@inproceedings{nguyen2024edit,
      title={Edit One for All: Interactive Batch Image Editing},
      author={Thao Nguyen and Utkarsh Ojha and Yuheng Li and Haotian Liu and Yong Jae Lee},
      year={2024},
      eprint={2401.10219},
      archivePrefix={arXiv},
      primaryClass={cs.CV}
   }