haofengl / dragnoise Goto Github PK

作者你好：
恭喜你喜提cvpr2024，论文的效果很好。我看完论文后，有两个想了解的点，想向你请教下哈哈。

1、模型耗时部分。
针对文章你提到的比DragDiffusion在点编辑的速度上快一倍，你有进行定量的比较过么，具体数值是啥呀？同时注意到你需要的Lora步数为200步，是否有其存在的原因，会增加不少耗时么？

2、点编辑和点跟踪是否到位？
DragDiffusion可能存在点编辑和点跟踪不到位的情况，具体表现在起点在完成编辑过程后，不能很好的跟进到设置终点的位置。请问你有做对应的优化或者可视化么？

祝生活愉悦~

Hello!
"1-LPIPS" and "Mean Distance": What is the specific size of these two indicators? I can't determine exactly in the diagram of the paper

(dragnoise) /content/DragNoise# python3 drag_ui.py
╭──────────── Traceback (most recent call last) ─────────────╮│ /content/DragNoise/drag_ui.py:4 in ││ │
│ 1 import os │
│ 2 import gradio as gr │
│ 3 │
│ ❱ 4 from utils.ui_utils import get_points, undo_points │
│ 5 from utils.ui_utils import clear_all, store_img, tra │
│ 6 from utils.ui_utils import clear_all_gen, store_img_ │
│ 7 │
│ │
│ /content/DragNoise/utils/ui_utils.py:25 in │
│ │
│ 22 │
│ 23 from .drag_utils import drag_diffusion_update │
│ 24 from .lora_utils import train_lora │
│ ❱ 25 from .attn_utils import register_attention_editor_di │
│ 26 from .freeu_utils import register_free_upblock2d, re │
│ 27 │
│ 28 │
╰────────────────────────────────────────────────────────────╯
ModuleNotFoundError: No module named 'utils.attn_utils'

你好，想请问有没有降低显存的办法，我想在12g的显存上跑，但是跑了几步后出现了out of memory

但是我看论文bottleneck feature是需要handle point 和target point进行更新的吧？请问有unet训练部分的代码么？