add highpass filter

AI4EPS · Sep 24, 2021 · ca9e29f · ca9e29f
1 parent eba5272
commit ca9e29f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/phasenet/data_reader.py b/phasenet/data_reader.py
@@ -175,6 +175,8 @@ def __init__(self, format="numpy", config=DataConfig(), **kwargs):
         self.label_width = config.label_width
         self.config = config
         self.format = format
+        if "highpass_filter" in kwargs:
+            self.highpass_filter = kwargs["highpass_filter"]
         if format in ["numpy", "mseed", "sac"]:
             self.data_dir = kwargs["data_dir"]
             try:
@@ -291,6 +293,8 @@ def read_mseed(self, fname):
         mseed = obspy.read(fname)
         mseed = mseed.detrend("spline", order=2, dspline=5 * mseed[0].stats.sampling_rate)
         mseed = mseed.merge(fill_value=0)
+        if self.highpass_filter > 0:
+            mseed = mseed.filter("highpass", freq=self.highpass_filter)
         starttime = min([st.stats.starttime for st in mseed])
         endtime = max([st.stats.endtime for st in mseed])
         mseed = mseed.trim(starttime, endtime, pad=True, fill_value=0)
@@ -326,6 +330,8 @@ def read_sac(self, fname, traces):
             mseed += obspy.read(tr, format="sac")
         mseed = mseed.detrend("spline", order=2, dspline=5 * mseed[0].stats.sampling_rate)
         mseed = mseed.merge(fill_value=0)
+        if self.highpass_filter > 0:
+            mseed = mseed.filter("highpass", freq=self.highpass_filter)
         starttime = min([st.stats.starttime for st in mseed])
         endtime = max([st.stats.endtime for st in mseed])
         mseed = mseed.trim(starttime, endtime, pad=True, fill_value=0)

diff --git a/phasenet/predict.py b/phasenet/predict.py
@@ -38,6 +38,7 @@ def read_args():
     parser.add_argument("--hdf5_group", default="data", help="data group name in hdf5 file")
     parser.add_argument("--result_dir", default="results", help="Output directory")
     parser.add_argument("--result_fname", default="picks", help="Output file")
+    parser.add_argument("--highpass_filter", default=0.0, type=float, help="Highpass filter")
     parser.add_argument("--min_p_prob", default=0.3, type=float, help="Probability threshold for P pick")
     parser.add_argument("--min_s_prob", default=0.3, type=float, help="Probability threshold for S pick")
     parser.add_argument("--mpd", default=50, type=float, help="Minimum peak distance")
@@ -170,6 +171,7 @@ def main(args):
                 hdf5_file=args.hdf5_file,
                 hdf5_group=args.hdf5_group,
                 amplitude=args.amplitude,
+                highpass_filter=args.highpass_filter,
             )
 
         pred_fn(args, data_reader, log_dir=args.result_dir)