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