DragGAN開(kāi)源三天Star量23k，這又來(lái)一個(gè)DragDiffusion

作者：機(jī)器之心 2023-06-28 13:48:23

動(dòng)動(dòng)鼠標(biāo)，讓圖片變「活」，成為你想要的模樣。

在 AIGC 的神奇世界里，我們可以在圖像上通過(guò)「拖曳」的方式，改變并合成自己想要的圖像。比如讓一頭獅子轉(zhuǎn)頭并張嘴：

實(shí)現(xiàn)這一效果的研究出自華人一作領(lǐng)銜的「Drag Your GAN」論文，于上個(gè)月放出并已被 SIGGRAPH 2023 會(huì)議接收。

一個(gè)多月過(guò)去了，該研究團(tuán)隊(duì)于近日放出了官方代碼。短短三天時(shí)間，Star 量便已突破了 23k，足可見(jiàn)其火爆程度。

圖片

GitHub 地址：https://github.com/XingangPan/DragGAN

無(wú)獨(dú)有偶，今日又一項(xiàng)類(lèi)似的研究 —— DragDiffusion 進(jìn)入了人們的視線。此前的 DragGAN 實(shí)現(xiàn)了基于點(diǎn)的交互式圖像編輯，并取得像素級(jí)精度的編輯效果。但是也有不足，DragGAN 是基于生成對(duì)抗網(wǎng)絡(luò)（GAN），通用性會(huì)受到預(yù)訓(xùn)練 GAN 模型容量的限制。

在新研究中，新加坡國(guó)立大學(xué)和字節(jié)跳動(dòng)的幾位研究者將這類(lèi)編輯框架擴(kuò)展到了擴(kuò)散模型，提出了 DragDiffusion。他們利用大規(guī)模預(yù)訓(xùn)練擴(kuò)散模型，極大提升了基于點(diǎn)的交互式編輯在現(xiàn)實(shí)世界場(chǎng)景中的適用性。

雖然現(xiàn)在大多數(shù)基于擴(kuò)散的圖像編輯方法都適用于文本嵌入，但 DragDiffusion 優(yōu)化了擴(kuò)散潛在表示，實(shí)現(xiàn)了精確的空間控制。

圖片

論文地址：https://arxiv.org/pdf/2306.14435.pdf
項(xiàng)目地址：https://yujun-shi.github.io/projects/dragdiffusion.html

研究者表示，擴(kuò)散模型以迭代方式生成圖像，而「一步」優(yōu)化擴(kuò)散潛在表示足以生成連貫結(jié)果，使 DragDiffusion 高效完成了高質(zhì)量編輯。

他們?cè)诟鞣N具有挑戰(zhàn)性的場(chǎng)景（如多對(duì)象、不同對(duì)象類(lèi)別）下進(jìn)行了廣泛實(shí)驗(yàn)，驗(yàn)證了 DragDiffusion 的可塑性和通用性。相關(guān)代碼也將很快放出、

下面我們看看 DragDiffusion 效果如何。

首先，我們想讓下圖中的小貓咪的頭再抬高一點(diǎn)，用戶(hù)只需將紅色的點(diǎn)拖拽至藍(lán)色的點(diǎn)就可以了：

接下來(lái)，我們想讓山峰變得再高一點(diǎn)，也沒(méi)有問(wèn)題，拖拽紅色關(guān)鍵點(diǎn)就可以了：

圖片

還想讓雕塑的頭像轉(zhuǎn)個(gè)頭，拖拽一下就能辦到：

圖片

讓岸邊的花，開(kāi)的范圍更廣一點(diǎn)：

方法介紹

本文提出的 DRAGDIFFUSION 旨在優(yōu)化特定的擴(kuò)散潛變量，以實(shí)現(xiàn)可交互的、基于點(diǎn)的圖像編輯。

為了實(shí)現(xiàn)這一目標(biāo)，該研究首先在擴(kuò)散模型的基礎(chǔ)上微調(diào) LoRA，以重建用戶(hù)輸入圖像。這樣做可以保證輸入、輸出圖像的風(fēng)格保持一致。

接下來(lái)，研究者對(duì)輸入圖像采用 DDIM inversion（這是一種探索擴(kuò)散模型的逆變換和潛在空間操作的方法），以獲得特定步驟的擴(kuò)散潛變量。

在編輯過(guò)程中，研究者反復(fù)運(yùn)用動(dòng)作監(jiān)督和點(diǎn)跟蹤，以?xún)?yōu)化先前獲得的第 t 步擴(kuò)散潛變量，從而將處理點(diǎn)的內(nèi)容「拖拽（drag）」到目標(biāo)位置。編輯過(guò)程還應(yīng)用了正則化項(xiàng)，以確保圖像的未掩碼區(qū)域保持不變。

最后，通過(guò) DDIM 對(duì)優(yōu)化后的第 t 步潛變量進(jìn)行去噪，得到編輯后的結(jié)果。總體概覽圖如下所示：

圖片