PhaseNet 0.2.2__tar.gz → 0.2.4__tar.gz

{phasenet-0.2.2/phasenet.egg-info → phasenet-0.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: phasenet
-Version: 0.2.2
+Version: 0.2.4
 Summary: A PyTorch implementation of PhaseNet for seismic and DAS phase picking
 Author: Weiqiang Zhu
 License-Expression: MIT

{phasenet-0.2.2 → phasenet-0.2.4}/phasenet/utils/visualization.py

@@ -194,9 +194,9 @@ def plot_phasenet_das_plus_train(
     imshow_kwargs = dict(aspect="auto", interpolation="nearest")
 
     for i in range(nb):
-        fig, axes = plt.subplots(2, 4, figsize=(20, 8))
+        fig, axes = plt.subplots(2, 2, figsize=(12, 8))
 
-        # Row 1: Raw Data | Phase Label | Event Center Label | Event Time Label
+        # Row 1: Raw Data | Phase Label
         raw_i = raw[i].T
         vmax_raw = np.percentile(np.abs(raw_i), 95) or 1.0
         axes[0, 0].imshow(raw_i, cmap="seismic", vmin=-vmax_raw, vmax=vmax_raw, **imshow_kwargs)
@@ -208,19 +208,7 @@ def plot_phasenet_das_plus_train(
             axes[0, 1].imshow(_das_phase_rgb(y_phase_pick[i], mask=mask_i), **imshow_kwargs)
         axes[0, 1].set_title("Phase Label")
 
-        cm = y_center_mask[i] if y_center_mask is not None else None
-        tm = y_time_mask[i] if y_time_mask is not None else None
-
-        if y_event_center is not None:
-            _plot_event_center(axes[0, 2], fig, y_event_center[i], center_mask=cm, imshow_kwargs=imshow_kwargs)
-        axes[0, 2].set_title("Event Center Label")
-
-        et_vlim = None
-        if y_event_time is not None and tm is not None:
-            et_vlim = _plot_event_time(axes[0, 3], fig, y_event_time[i].T, tm.T, imshow_kwargs)
-        axes[0, 3].set_title("Event Time Label")
-
-        # Row 2: Normalized Data | Phase Pred | Event Center Pred | Event Time Pred
+        # Row 2: Normalized Data | Phase Pred
         norm_i = norm_data[i].T
         vmax_norm = np.percentile(np.abs(norm_i), 95) or 1.0
         axes[1, 0].imshow(norm_i, cmap="seismic", vmin=-vmax_norm, vmax=vmax_norm, **imshow_kwargs)
@@ -231,14 +219,6 @@ def plot_phasenet_das_plus_train(
             axes[1, 1].imshow(_das_phase_rgb(phase_pick[i]), **imshow_kwargs)
         axes[1, 1].set_title("Phase Pred")
 
-        if event_center is not None:
-            _plot_event_center(axes[1, 2], fig, event_center[i], center_mask=None, imshow_kwargs=imshow_kwargs)
-        axes[1, 2].set_title("Event Center Pred")
-
-        if event_time is not None and tm is not None:
-            _plot_event_time(axes[1, 3], fig, event_time[i].T, tm.T, imshow_kwargs, vlim=et_vlim)
-        axes[1, 3].set_title("Event Time Pred")
-
         for ax in axes.flat:
             ax.grid(True, alpha=0.3, linewidth=0.5)
         for ax in axes[1]:

{phasenet-0.2.2 → phasenet-0.2.4/phasenet.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: phasenet
-Version: 0.2.2
+Version: 0.2.4
 Summary: A PyTorch implementation of PhaseNet for seismic and DAS phase picking
 Author: Weiqiang Zhu
 License-Expression: MIT

phasenet-0.2.4/phasenet.egg-info/entry_points.txt

@@ -0,0 +1,3 @@
+[console_scripts]
+phasenet-predict = predict:cli
+phasenet-train = train:cli

{phasenet-0.2.2 → phasenet-0.2.4}/predict.py

@@ -625,6 +625,10 @@ def get_args_parser(add_help=True):
     return parser
 
 
-if __name__ == "__main__":
+def cli():
     args = get_args_parser().parse_args()
     main(args)
+
+
+if __name__ == "__main__":
+    cli()

{phasenet-0.2.2 → phasenet-0.2.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "phasenet"
-version = "0.2.2"
+version = "0.2.4"
 description = "A PyTorch implementation of PhaseNet for seismic and DAS phase picking"
 readme = "README.md"
 license = "MIT"
@@ -41,8 +41,8 @@ Homepage = "https://github.com/AI4EPS/phasenet-pytorch"
 Repository = "https://github.com/AI4EPS/phasenet-pytorch"
 
 [project.scripts]
-phasenet-train = "train:main"
-phasenet-predict = "predict:main"
+phasenet-train = "train:cli"
+phasenet-predict = "predict:cli"
 
 [tool.setuptools]
 py-modules = ["train", "predict", "utils"]

{phasenet-0.2.2 → phasenet-0.2.4}/train.py

@@ -802,6 +802,10 @@ def get_args_parser(add_help=True):
     return parser
 
 
-if __name__ == "__main__":
+def cli():
     args = get_args_parser().parse_args()
     main(args)
+
+
+if __name__ == "__main__":
+    cli()

phasenet-0.2.2/phasenet.egg-info/entry_points.txt

@@ -1,3 +0,0 @@
-[console_scripts]
-phasenet-predict = predict:main
-phasenet-train = train:main