跳到主要內容
AI News HubLIVE
站內改寫2 分鐘閱讀

SelectAnyTree:一種用於3D森林LiDAR點雲的可提示例項分割模型

文章摘要

SelectAnyTree是一種基於提示的例項分割模型,旨在從3D森林LiDAR點雲中透過少量點選分割出任意單棵樹。該模型引入點選到查詢的提示編碼器和冠層高度模型引導的首次提示,結合狀態空間查詢解碼器,實現了高效的長距離上下文捕獲。在七個不同森林區域和獨立測試集上的評估表明,單次點選即可達到78.2的交併比,顯著優於現有方法。

來源arXiv Computer Vision作者: Trung Thanh Nguyen, Daniel Lusk, Kilian Gerberding, Janusch Vajna-Jehle, Tuan-Anh Vu, Duc Viet Le, Tu Vo, Phi Le Nguyen, Yasutomo Kawanishi, Takahiro Komamizu, Ichiro Ide, Julian Frey, Teja Kattenborn
SelectAnyTree:一種用於3D森林LiDAR點雲的可提示例項分割模型
回報錯誤

更正管道尚未開通,可先複製下方文章資訊留存。

查看更正說明
直接讀正文

隨著森林監測向可擴充套件、高精度的三維測量方向發展,自動化森林LiDAR點雲例項分割的重要性日益凸顯。然而,由於樹例項標籤的稀缺性,這一領域的進展受到嚴重製約。每公頃森林可能包含數百萬個點和數百個重疊且複雜的樹冠,這使得從原始資料手動標註成為一項既費時又容易出錯的工作。當前的標註方式通常基於自動預分割結果進行人工修正,但這種方法仍然成本高昂,且缺乏互動式或AI輔助的精細化功能。受基礎分割模型可提示正規化的啟發,研究人員提出了SelectAnyTree,這是一種可提示的例項分割模型,能夠透過少量點選在3D森林點雲中精確描繪出任意單棵樹。

SelectAnyTree的核心創新包括兩個關鍵元件:點選到查詢的提示編碼器(Click-to-query Prompt Encoder)和冠層高度模型(CHM)引導的首次提示。前者將每次使用者點選轉換為一個獨立的內容查詢,編碼其3D位置和正/負極性,並與池化的區域性骨幹特徵相結合。後者則在沒有使用者輸入的情況下,利用幾何和生態知識自動提供樹梢作為首次提示,從而引導模型關注目標樹冠。隨後,提示查詢透過一個狀態空間查詢解碼器被解碼為單個樹掩碼,該解碼器以線性時間複雜度高效捕獲大規模森林場景中的長距離上下文資訊。

研究團隊在七個不同森林區域和一個獨立保留測試資料集上,對SelectAnyTree進行了互動式和例項級別的評估。實驗結果充分展示了模型超越訓練域的強泛化能力。單次點選即可分割目標樹達到78.2的交併比(IoU),比最強的可提示基線高出24.8個百分點。此外,SelectAnyTree在達到每個精度目標時所需的點選次數最少,同時使用的引數數量和推理時間遠少於先前的可提示模型。該模型的原始碼已在GitHub上公開,供研究人員進一步使用和開發。

該模型的提出有望大幅降低森林點雲標註的成本和難度,為森林生態監測、碳儲量估算等應用提供高效的技術支撐。未來,團隊計劃進一步最佳化模型對異質森林型別的適應能力,並探索與無人機載LiDAR系統的整合。

展開要點與分析

文章情報

工程師進階

要點

  • 提出可提示模型SelectAnyTree,透過少量點選即可分割單棵樹。
  • 創新點包括點選到查詢編碼器和冠層高度模型引導的首次提示。
  • 使用狀態空間查詢解碼器處理大規模森林場景,線性時間複雜度。
  • 在多個區域測試中,單次點選IoU達78.2,引數和推理時間更少。

要點與分析由自動化流程生成,可能有誤,請結合原始來源核實。