seisbench/seisbench

SeisBench - A toolbox for machine learning in seismology

What it solves

SeisBench 減少了開發與應用機器學習技術於地震學所產生的額外負擔。它提供一種標準化的方式來存取地震資料集,並提供統一的 API 以在地震資料上訓練與部署 ML 模型,免除每個新專案都必須自行建構資料管線的需求。

How it works

此工具箱分為三個核心模組:

  • data:提供統一介面以載入與存取基準地震資料集。
  • models:包含一系列用於地震學的機器學習模型,使用者可建立新模型、載入預訓練模型,或在特定資料集上訓練現有模型。
  • generate:透過提供建構資料生成管線的工具,彌合資料與模型之間的鴻溝。

Who it’s for

適用於需要處理地震資料、訓練模型以完成如地震波形挑選、去噪或深度估計等任務,並且使用分散式聲學感測(DAS)資料的地震學家與機器學習研究者。

Highlights

  • Unified API:簡化存取資料集與套用 ML 演算法的流程。
  • Pretrained Models:提供一系列可直接使用的地震學任務模型。
  • DAS Support:包含針對分散式聲學感測資料的專屬工具與模型。
  • Comprehensive Examples:提供多種任務的 Colab notebook,包括事件目錄建構與去噪。