├── .gitignore ├── .pre-commit-config.yaml ├── LICENSE ├── README.md ├── app.py ├── assets ├── accurate_rabbit.jpg ├── accurate_rabbit_result.jpg ├── canny.jpg ├── canny_result.jpg ├── control_input.jpg ├── depth.jpg ├── depth_result.jpg ├── gradio_objremoval.jpg ├── gradio_objremoval_result.jpg ├── gradio_outpaint.jpg ├── gradio_outpaint_result.jpg ├── gradio_text_objinpaint.jpg ├── gradio_text_objinpaint_result.jpg ├── hed.jpg ├── hed_result.jpg ├── pose_control.jpg ├── pose_input.jpg ├── pose_result.jpg ├── shapeguided_s1.jpg ├── shapeguided_s1_rabbit.jpg ├── shapeguided_s1_rabbit_high.jpg ├── shapeguided_s1_rabbit_high_result.jpg ├── shapeguided_s1_rabbit_result.jpg └── shapeguided_s1_result.jpg ├── configs ├── acc.yaml ├── ppt1_sd15.yaml └── ppt2_bn.yaml ├── examples └── cake.jpg ├── powerpaint ├── __init__.py ├── datasets │ ├── __init__.py │ ├── laion.py │ └── openimage.py ├── models │ ├── __init__.py │ ├── brushnet.py │ ├── unet_2d_blocks.py │ └── unet_2d_condition.py ├── pipelines │ ├── __init__.py │ ├── pipeline_powerpaint.py │ ├── pipeline_powerpaint_brushnet.py │ └── pipeline_powerpaint_controlnet.py └── utils │ ├── __init__.py │ └── loaders.py ├── pyproject.toml ├── requirements ├── ppt.yml └── requirements.txt ├── submit.py ├── train_ppt1_sd15.py └── train_ppt2_bn.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/README.md -------------------------------------------------------------------------------- /app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/app.py -------------------------------------------------------------------------------- /assets/accurate_rabbit.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/assets/accurate_rabbit.jpg -------------------------------------------------------------------------------- /assets/accurate_rabbit_result.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/assets/accurate_rabbit_result.jpg -------------------------------------------------------------------------------- /assets/canny.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/assets/canny.jpg -------------------------------------------------------------------------------- /assets/canny_result.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/assets/canny_result.jpg -------------------------------------------------------------------------------- /assets/control_input.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/assets/control_input.jpg -------------------------------------------------------------------------------- /assets/depth.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/assets/depth.jpg -------------------------------------------------------------------------------- /assets/depth_result.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/assets/depth_result.jpg -------------------------------------------------------------------------------- /assets/gradio_objremoval.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/assets/gradio_objremoval.jpg -------------------------------------------------------------------------------- /assets/gradio_objremoval_result.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/assets/gradio_objremoval_result.jpg -------------------------------------------------------------------------------- /assets/gradio_outpaint.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/assets/gradio_outpaint.jpg -------------------------------------------------------------------------------- /assets/gradio_outpaint_result.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/assets/gradio_outpaint_result.jpg -------------------------------------------------------------------------------- /assets/gradio_text_objinpaint.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/assets/gradio_text_objinpaint.jpg -------------------------------------------------------------------------------- /assets/gradio_text_objinpaint_result.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/assets/gradio_text_objinpaint_result.jpg -------------------------------------------------------------------------------- /assets/hed.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/assets/hed.jpg -------------------------------------------------------------------------------- /assets/hed_result.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/assets/hed_result.jpg -------------------------------------------------------------------------------- /assets/pose_control.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/assets/pose_control.jpg -------------------------------------------------------------------------------- /assets/pose_input.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/assets/pose_input.jpg -------------------------------------------------------------------------------- /assets/pose_result.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/assets/pose_result.jpg -------------------------------------------------------------------------------- /assets/shapeguided_s1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/assets/shapeguided_s1.jpg -------------------------------------------------------------------------------- /assets/shapeguided_s1_rabbit.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/assets/shapeguided_s1_rabbit.jpg -------------------------------------------------------------------------------- /assets/shapeguided_s1_rabbit_high.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/assets/shapeguided_s1_rabbit_high.jpg -------------------------------------------------------------------------------- /assets/shapeguided_s1_rabbit_high_result.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/assets/shapeguided_s1_rabbit_high_result.jpg -------------------------------------------------------------------------------- /assets/shapeguided_s1_rabbit_result.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/assets/shapeguided_s1_rabbit_result.jpg -------------------------------------------------------------------------------- /assets/shapeguided_s1_result.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/assets/shapeguided_s1_result.jpg -------------------------------------------------------------------------------- /configs/acc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/configs/acc.yaml -------------------------------------------------------------------------------- /configs/ppt1_sd15.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/configs/ppt1_sd15.yaml -------------------------------------------------------------------------------- /configs/ppt2_bn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/configs/ppt2_bn.yaml -------------------------------------------------------------------------------- /examples/cake.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/examples/cake.jpg -------------------------------------------------------------------------------- /powerpaint/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/powerpaint/__init__.py -------------------------------------------------------------------------------- /powerpaint/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/powerpaint/datasets/__init__.py -------------------------------------------------------------------------------- /powerpaint/datasets/laion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/powerpaint/datasets/laion.py -------------------------------------------------------------------------------- /powerpaint/datasets/openimage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/powerpaint/datasets/openimage.py -------------------------------------------------------------------------------- /powerpaint/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/powerpaint/models/__init__.py -------------------------------------------------------------------------------- /powerpaint/models/brushnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/powerpaint/models/brushnet.py -------------------------------------------------------------------------------- /powerpaint/models/unet_2d_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/powerpaint/models/unet_2d_blocks.py -------------------------------------------------------------------------------- /powerpaint/models/unet_2d_condition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/powerpaint/models/unet_2d_condition.py -------------------------------------------------------------------------------- /powerpaint/pipelines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/powerpaint/pipelines/__init__.py -------------------------------------------------------------------------------- /powerpaint/pipelines/pipeline_powerpaint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/powerpaint/pipelines/pipeline_powerpaint.py -------------------------------------------------------------------------------- /powerpaint/pipelines/pipeline_powerpaint_brushnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/powerpaint/pipelines/pipeline_powerpaint_brushnet.py -------------------------------------------------------------------------------- /powerpaint/pipelines/pipeline_powerpaint_controlnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/powerpaint/pipelines/pipeline_powerpaint_controlnet.py -------------------------------------------------------------------------------- /powerpaint/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/powerpaint/utils/__init__.py -------------------------------------------------------------------------------- /powerpaint/utils/loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/powerpaint/utils/loaders.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements/ppt.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/requirements/ppt.yml -------------------------------------------------------------------------------- /requirements/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/requirements/requirements.txt -------------------------------------------------------------------------------- /submit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/submit.py -------------------------------------------------------------------------------- /train_ppt1_sd15.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/train_ppt1_sd15.py -------------------------------------------------------------------------------- /train_ppt2_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/open-mmlab/PowerPaint/HEAD/train_ppt2_bn.py --------------------------------------------------------------------------------