├── LICENSE ├── README.md ├── assets ├── GoodDrag_demo.gif ├── cat_2 │ ├── image_with_new_points.png │ ├── image_with_points.jpg │ ├── original.jpg │ └── trajectory.gif ├── chess_1 │ ├── image_with_new_points.png │ ├── image_with_points.jpg │ ├── original.jpg │ └── trajectory.gif ├── furniture_0 │ ├── image_with_new_points.png │ ├── image_with_points.jpg │ ├── original.jpg │ └── trajectory.gif ├── gooddrag_icon.png ├── human_6 │ ├── image_with_new_points.png │ ├── image_with_points.jpg │ ├── original.jpg │ └── trajectory.gif ├── leopard │ ├── image_with_new_points.png │ ├── image_with_points.jpg │ ├── original.jpg │ └── trajectory.gif └── rabbit │ ├── image_with_new_points.png │ ├── image_with_points.jpg │ ├── original.jpg │ └── trajectory.gif ├── bench_gooddrag.py ├── dataset ├── cat_2 │ ├── image_with_points.jpg │ ├── mask.png │ ├── original.jpg │ └── points.json └── furniture_0 │ ├── image_with_points.jpg │ ├── mask.png │ ├── original.jpg │ └── points.json ├── environment.yaml ├── evaluation ├── GScore.ipynb └── compute_DAI.py ├── gooddrag_ui.py ├── pipeline.py ├── requirements.txt ├── utils ├── __pycache__ │ ├── attn_utils.cpython-310.pyc │ ├── drag_utils.cpython-310.pyc │ ├── lora_utils.cpython-310.pyc │ └── ui_utils.cpython-310.pyc ├── attn_utils.py ├── drag_utils.py ├── lora_utils.py └── ui_utils.py ├── webui.bat └── webui.sh /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/README.md -------------------------------------------------------------------------------- /assets/GoodDrag_demo.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/assets/GoodDrag_demo.gif -------------------------------------------------------------------------------- /assets/cat_2/image_with_new_points.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/assets/cat_2/image_with_new_points.png -------------------------------------------------------------------------------- /assets/cat_2/image_with_points.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/assets/cat_2/image_with_points.jpg -------------------------------------------------------------------------------- /assets/cat_2/original.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/assets/cat_2/original.jpg -------------------------------------------------------------------------------- /assets/cat_2/trajectory.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/assets/cat_2/trajectory.gif -------------------------------------------------------------------------------- /assets/chess_1/image_with_new_points.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/assets/chess_1/image_with_new_points.png -------------------------------------------------------------------------------- /assets/chess_1/image_with_points.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/assets/chess_1/image_with_points.jpg -------------------------------------------------------------------------------- /assets/chess_1/original.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/assets/chess_1/original.jpg -------------------------------------------------------------------------------- /assets/chess_1/trajectory.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/assets/chess_1/trajectory.gif -------------------------------------------------------------------------------- /assets/furniture_0/image_with_new_points.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/assets/furniture_0/image_with_new_points.png -------------------------------------------------------------------------------- /assets/furniture_0/image_with_points.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/assets/furniture_0/image_with_points.jpg -------------------------------------------------------------------------------- /assets/furniture_0/original.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/assets/furniture_0/original.jpg -------------------------------------------------------------------------------- /assets/furniture_0/trajectory.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/assets/furniture_0/trajectory.gif -------------------------------------------------------------------------------- /assets/gooddrag_icon.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/assets/gooddrag_icon.png -------------------------------------------------------------------------------- /assets/human_6/image_with_new_points.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/assets/human_6/image_with_new_points.png -------------------------------------------------------------------------------- /assets/human_6/image_with_points.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/assets/human_6/image_with_points.jpg -------------------------------------------------------------------------------- /assets/human_6/original.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/assets/human_6/original.jpg -------------------------------------------------------------------------------- /assets/human_6/trajectory.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/assets/human_6/trajectory.gif -------------------------------------------------------------------------------- /assets/leopard/image_with_new_points.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/assets/leopard/image_with_new_points.png -------------------------------------------------------------------------------- /assets/leopard/image_with_points.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/assets/leopard/image_with_points.jpg -------------------------------------------------------------------------------- /assets/leopard/original.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/assets/leopard/original.jpg -------------------------------------------------------------------------------- /assets/leopard/trajectory.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/assets/leopard/trajectory.gif -------------------------------------------------------------------------------- /assets/rabbit/image_with_new_points.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/assets/rabbit/image_with_new_points.png -------------------------------------------------------------------------------- /assets/rabbit/image_with_points.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/assets/rabbit/image_with_points.jpg -------------------------------------------------------------------------------- /assets/rabbit/original.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/assets/rabbit/original.jpg -------------------------------------------------------------------------------- /assets/rabbit/trajectory.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/assets/rabbit/trajectory.gif -------------------------------------------------------------------------------- /bench_gooddrag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/bench_gooddrag.py -------------------------------------------------------------------------------- /dataset/cat_2/image_with_points.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/dataset/cat_2/image_with_points.jpg -------------------------------------------------------------------------------- /dataset/cat_2/mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/dataset/cat_2/mask.png -------------------------------------------------------------------------------- /dataset/cat_2/original.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/dataset/cat_2/original.jpg -------------------------------------------------------------------------------- /dataset/cat_2/points.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/dataset/cat_2/points.json -------------------------------------------------------------------------------- /dataset/furniture_0/image_with_points.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/dataset/furniture_0/image_with_points.jpg -------------------------------------------------------------------------------- /dataset/furniture_0/mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/dataset/furniture_0/mask.png -------------------------------------------------------------------------------- /dataset/furniture_0/original.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/dataset/furniture_0/original.jpg -------------------------------------------------------------------------------- /dataset/furniture_0/points.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/dataset/furniture_0/points.json -------------------------------------------------------------------------------- /environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/environment.yaml -------------------------------------------------------------------------------- /evaluation/GScore.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/evaluation/GScore.ipynb -------------------------------------------------------------------------------- /evaluation/compute_DAI.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/evaluation/compute_DAI.py -------------------------------------------------------------------------------- /gooddrag_ui.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/gooddrag_ui.py -------------------------------------------------------------------------------- /pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/pipeline.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/requirements.txt -------------------------------------------------------------------------------- /utils/__pycache__/attn_utils.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/utils/__pycache__/attn_utils.cpython-310.pyc -------------------------------------------------------------------------------- /utils/__pycache__/drag_utils.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/utils/__pycache__/drag_utils.cpython-310.pyc -------------------------------------------------------------------------------- /utils/__pycache__/lora_utils.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/utils/__pycache__/lora_utils.cpython-310.pyc -------------------------------------------------------------------------------- /utils/__pycache__/ui_utils.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/utils/__pycache__/ui_utils.cpython-310.pyc -------------------------------------------------------------------------------- /utils/attn_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/utils/attn_utils.py -------------------------------------------------------------------------------- /utils/drag_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/utils/drag_utils.py -------------------------------------------------------------------------------- /utils/lora_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/utils/lora_utils.py -------------------------------------------------------------------------------- /utils/ui_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/utils/ui_utils.py -------------------------------------------------------------------------------- /webui.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/webui.bat -------------------------------------------------------------------------------- /webui.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zewei-Zhang/GoodDrag/HEAD/webui.sh --------------------------------------------------------------------------------