首頁  >  文章  >  科技週邊  >  AI繪圖模型訓練到推理,一個網頁就能搞定

AI繪圖模型訓練到推理,一個網頁就能搞定

WBOY
WBOY轉載
2024-02-19 09:24:281023瀏覽

不會寫程式碼,也能訓練AI繪畫模型了!

只要藉助這個框架,從訓練到推理都能一站式搞定,還能一次管理多個模型。

AI繪圖模型訓練到推理,一個網頁就能搞定

阿里巴巴團隊推出並開源了這款萬用圖片產生工作台SCEPTER Studio。

有了它,不用程式碼,直接在Web介面當中就能完成模型的訓練與微調,並管理相關資料。

團隊也推出了內建三種型號的DEMO,可以在線上體驗SCEPTER的推理功能。

那麼SCEPTER具體都能幹啥,下面就來一睹為快!

一站式管理繪圖模型

使用SCEPTER,編寫程式變得不再必要。只要在Web頁面中選擇模型並調整參數,就能完成訓練和微調。

AI繪圖模型訓練到推理,一個網頁就能搞定

具體來看,模型方面,SCEPTER目前支援Stable Diffusion系列的SD 1.5/2.1和SDXL模型。

微調方式上,它支援傳統的全量微調、LoRA等方式,以及自家的SCEdit微調框架,未來還會加入對Res-Tuning調優方式的支援。

SCEdit透過跳過連接的方式,讓擴散模​​型能以更高的效率完成圖片生成任務,相比LoRA節約了30%-50%的記憶體開銷。

訓練資料方面,SCEPTER內建了包含3D、日漫、油畫、素描等6種風格,每種風格30組圖像-文字對的自創資料集。

AI繪圖模型訓練到推理,一個網頁就能搞定

同時,也可將自行準備的圖片打包壓縮,並以csv或txt文件建立影像(檔案名稱)和prompt的對應關係,匯入SCEPTER平台。

AI繪圖模型訓練到推理,一個網頁就能搞定

如果連文件也不想寫,也可以在SCEPTER介面中直接上傳圖片並標註Prompt,管理、新增、刪除資料集中的圖片。

AI繪圖模型訓練到推理,一個網頁就能搞定

到了推理階段,SCEPTER支援的下游任務包括文生圖和可控影像合成,未來也將支援影像編輯,用法和現有的Web版SD相似。

AI繪圖模型訓練到推理,一個網頁就能搞定

同時,SCEPTER介面中也整合了咒語書(Prompt集合)和一些現成的微調模型。

AI繪圖模型訓練到推理,一個網頁就能搞定

那麼,SCEPTER該如何食用呢?

如果只想玩玩生成的話,官方在HuggingFace和魔搭社群上的DEMO就可以滿足要求,並且後者還是中文介面。

如果要使用資料管理、訓練等功能,就需要自行安裝部署完整版本了,具體步驟可以參考GitHub頁面中的教學。

整個過程中,只有安裝部署的環節需要用到一些簡單的程式碼,之後的一切過程就都可以在Web介面裡直接操作了。

有興趣的話,就去體驗一下吧!

傳送門:https://github.com/modelscope/scepter

以上是AI繪圖模型訓練到推理,一個網頁就能搞定的詳細內容。更多資訊請關注PHP中文網其他相關文章!

陳述:
本文轉載於:51cto.com。如有侵權,請聯絡admin@php.cn刪除