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,包括事件目录构建与去噪。