Skip to main content

Deep learning seismic phase picking package

Project description

# SeisNN

Seismic event P phase picking project

Main framework: Obspy, Seisan, Tensorflow with Keras, Pytorch

Using U-net to generate pick probability


Early access

The code is still in the development state, API will change frequently.

Please star us for upcoming updates!


  • S-File catalog from [SEISAN](

  • SeisComP Data Structure (SDS) database. The directory and file layout of SDS is defined as:



  • pip install seisnn
  • Clone this repository
  • Follow the instructions in the [Docker](docker) folder to create a Docker container.
  • SSH into the Docker container you create.

In the [scripts](scripts) folder:

  • [](scripts/ - Make S-Files into training .pkl datasets.
  • [](seisnn/ - Calculate precision, recall and F1 score. - Plot error distribution.
  • [](scripts/ - Plot the wavefile, picks and the probability form the .pkl dataset.
  • [](scripts/ - Scan through all stations available in the given time window, transfer into .pkl dataset.


You can choose between [scripts/tensorflow](scripts/tensorflow) or [scripts/pytorch](scripts/pytorch)

  • [](scripts/tensorflow/ - Pre-train the model using small dataset.
  • [](scripts/tensorflow/ - Train the model with the pre-trained weight.
  • [](scripts/tensorflow/ - Predict the probability of the picks and write into the dataset.


The model is stored in [seisnn/tensorflow](seisnn/tensorflow) or [seisnn/pytorch](seisnn/pytorch)



Zhu, W., & Beroza, G. C. (2018). PhaseNet: A Deep-Neural-Network-Based Seismic Arrival Time Picking Method. arXiv preprint arXiv:1803.03211.


Ronneberger, O., Fischer, P., & Brox, T. (2015, October). U-net: Convolutional networks for biomedical image segmentation. In International Conference on Medical image computing and computer-assisted intervention (pp. 234-241). Springer, Cham.

[U-net ++](

Zhou, Z., Siddiquee, M. M. R., Tajbakhsh, N., & Liang, J. (2018). Unet++: A nested u-net architecture for medical image segmentation. In Deep Learning in Medical Image Analysis and Multimodal Learning for Clinical Decision Support (pp. 3-11). Springer, Cham.

[Jimmy Lab wordpress](


Project details

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Files for SeisNN, version 0.0.2.dev1
Filename, size File type Python version Upload date Hashes
Filename, size SeisNN-0.0.2.dev1-py3-none-any.whl (12.1 kB) File type Wheel Python version py3 Upload date Hashes View
Filename, size SeisNN-0.0.2.dev1.tar.gz (9.5 kB) File type Source Python version None Upload date Hashes View

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring DigiCert DigiCert EV certificate Facebook / Instagram Facebook / Instagram PSF Sponsor Fastly Fastly CDN Google Google Object Storage and Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Salesforce Salesforce PSF Sponsor Sentry Sentry Error logging StatusPage StatusPage Status page