用於使用 Segment Anything Model (SAM) 分割地理空間資料的 Python 套件
Segment-Geospatial套件的靈感來自 Aliaksandr Hancharenka 創作的 Segment-Anything-eo 儲存庫。為了方便對地理空間資料使用分段任意模型 (SAM),我開發了分段-anything-py 和分段-地理空間 Python 包,這些包現在可以在 PyPI 和 conda-forge 上使用。我的主要目標是簡化利用 SAM 進行地理空間資料分析的過程,使用戶能夠以最少的編碼工作來實現這一目標。我從segment-anything-eo 儲存庫中改編了segment-geospatial 的源代碼,其原始版本歸功於Aliaksandr Hancharenka。
分段地理空間可在 PyPI 上使用。若要安裝segment-geospatial ,請在終端機中執行以下命令:
pip install segment-geospatial
分段地理空間也可在 conda-forge 上使用。如果您的電腦上安裝了 Anaconda 或 Miniconda,則可以使用以下命令安裝segment-geospatial。建議為segment-geospatial建立一個新的 conda 環境。以下命令將建立一個名為geo
新conda環境並安裝segment-geospatial及其相依性:
conda create -n geo python
conda activate geo
conda install -c conda-forge mamba
mamba install -c conda-forge segment-geospatial
如果您的系統有GPU,但上述指令沒有安裝GPU版本的pytorch,您可以使用下列指令強制安裝GPU版本的pytorch:
mamba install -c conda-forge segment-geospatial " pytorch=*=cuda* "
Samgeo-geospatial 有一些可選的依賴項,這些依賴項未包含在預設的 conda 環境中。若要安裝這些依賴項,請執行下列命令:
mamba install -c conda-forge groundingdino-py segment-anything-fast
我的 YouTube 頻道上提供了影片教學。
Segment Anything Model 是計算密集型的,建議使用強大的 GPU 來處理大型資料集。建議使用具有至少 8 GB GPU 記憶體的 GPU。您可以利用 Google Colab 提供的免費 GPU 資源。或者,您可以申請 AWS Cloud Credit for Research,該計畫提供雲端積分來支援學術研究。如果您位於大中華區,請在此申請 AWS 雲端積分。
此存儲庫及其內容僅供教育目的。透過使用所提供的資訊和程式碼,使用者承認他們使用 API 和模型的風險由其自行承擔,並同意遵守任何適用的法律和法規。建議打算從任何底圖下載大量圖像切片的用戶在這樣做之前聯繫底圖提供者以獲得許可。未經授權使用底圖或其任何組件可能會違反版權法或其他適用的法律和法規。
請參閱貢獻指南以獲取更多資訊。
該計畫基於美國國家航空暨太空總署 (NASA) 的部分支持,撥款編號為 80NSSC22K1742,透過開源工具、框架和函式庫 2020 計畫頒發。
該計畫也得到亞馬遜網路服務(AWS)的支持。此外,該軟體包是透過以下開源專案實現的。這些項目的開發商都獲得了榮譽。