Meta AI 開源萬物可分割 AI 模型(SAM)

努力的小雨發表於2023-04-11

開始

4 月 6 日,根據 Meta AI 官方部落格,Meta AI 宣佈推出了一個 AI 模型 Segment Anything Model(SAM,分割一切模型)。據介紹,該模型能夠根據文字指令等方式實現影像分割,而且萬物皆可識別和一鍵摳圖。

github原始碼地址:facebookresearch/segment-anything

官方網站體驗地址:segment-anything.com/demo

特點

Segment Anything Model (SAM) 根據輸入提示(如點或框)生成高質量的物件蒙版,並可用於為影像中的所有物件生成蒙版。它已經在 11 萬張影像和 1 億個掩碼的資料集上進行了訓練,並且在各種分割任務上具有強大的零鏡頭效能。說到底是一種通用的影像分割方法。

初體驗

進入官網體驗地址後:點選同意條款和條件,大概意思是:

  1. 這是一個研究演示,不得用於任何商業目的
  2. 上傳的任何影像將僅用於演示分段任何模型。所有影像和從中派生的任何資料將在會話結束時刪除。
  3. 上傳的任何圖片均不得違反任何智慧財產權或 Facebook 社群守則。

進入上傳介面:

點選“Upload an image”,上傳自己的圖片:(也可以直接使用官方的圖片測試)
image

我上傳完圖片用時3分11秒,圖片大小隻有50多K。還有一張4M多的用時幾乎差不太多,看來跟圖片大小關係不是太大

Hover & Click——選取物體

基本操作:左鍵選擇物體,右鍵移除選取
image

選取完之後,可以使用Cut out object,或則再點選Multi-mask再次點選選擇標記點,可多次分割。

選完之後點選“Cut out object”,然後結果就儲存再“Cut-Outs”一欄中。

Box——框選物體

基本操作:按住左鍵選出一個框框範圍
image

儲存所選區域的方法仍然是點選“Cut out object”。然後結果就儲存再“Cut-Outs”一欄中。

Everything——分割所有物體

不用我們們操作了,直接交給AI處理,提取出所有可分割圖形物體
image

所有物體的區域都儲存在“Cut-Outs”一欄中:

Cut-Outs——結果提取

基本操作:只需要對Cut-Outs一欄的圖片右鍵點選,並在彈出的選單中選擇“將圖片另存為”即可。
image

結語

用起來感覺效果不錯,畢竟本人不是做ps或者圖片處理的,對圖片的質量要求不算高,不過Meta開源了segment-anything框架,屬實佩服,畢竟連資料集也可以進行下載,下載地址在github主頁下方,目前已經21k+star。希望AI可以繼續發展,讓「玩具」變成工具
公眾號

相關文章