forked from open-mmlab/mmagic
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[Enhance] Improve powerpaint (open-mmlab#2080)
* add readme * add readme * add require * fix readme * fix according to readme * fix lint
- Loading branch information
Showing
4 changed files
with
86 additions
and
3 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,60 @@ | ||
# A Task is Worth One Word: Learning with Task Prompts for High-Quality Versatile Image Inpainting | ||
|
||
### [Project Page](https://powerpaint.github.io/) | [Paper](https://arxiv.org/abs/2312.03594) | ||
|
||
This README provides a step-by-step guide to download the repository, set up the required virtual environment named "PowerPaint" using conda, and run PowerPaint with or without ControlNet. | ||
|
||
**Stronger Model Weights and Online Demo Coming Soon!** | ||
|
||
<img src='https://github.com/open-mmlab/mmagic/assets/12782558/acd01391-c73f-4997-aafd-0869aebcc915'/> | ||
|
||
## Getting Started | ||
|
||
```bash | ||
# Clone the Repository | ||
git clone https://github.com/open-mmlab/mmagic.git | ||
|
||
# Navigate to the Repository | ||
cd projects/powerpaint | ||
|
||
# Create Virtual Environment with Conda | ||
conda create --name PowerPaint python=3.9 | ||
conda activate PowerPaint | ||
|
||
# Install Dependencies | ||
pip install -r requirements.txt | ||
|
||
# Create Models Folder | ||
mkdir models | ||
|
||
# Set up Git LFS | ||
git lfs install | ||
|
||
# Clone PowerPaint Model | ||
git lfs clone https://huggingface.co/JunhaoZhuang/PowerPaint-v1/ ./models | ||
``` | ||
|
||
## Run PowerPaint | ||
|
||
To run PowerPaint, execute the following command: | ||
|
||
```bash | ||
python gradio_PowerPaint.py | ||
``` | ||
|
||
This command will launch the Gradio interface for PowerPaint. | ||
|
||
Feel free to explore and create stunning images with PowerPaint! | ||
|
||
## BibTeX | ||
|
||
``` | ||
@misc{zhuang2023task, | ||
title={A Task is Worth One Word: Learning with Task Prompts for High-Quality Versatile Image Inpainting}, | ||
author={Junhao Zhuang and Yanhong Zeng and Wenran Liu and Chun Yuan and Kai Chen}, | ||
year={2023}, | ||
eprint={2312.03594}, | ||
archivePrefix={arXiv}, | ||
primaryClass={cs.CV} | ||
} | ||
``` |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,8 @@ | ||
controlnet-aux==0.0.3 | ||
diffusers==0.23.1 | ||
gradio==3.23.0 | ||
mmengine | ||
opencv-python | ||
torch | ||
torchvision | ||
transformers |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters