phasenet 0.2.6.dev2__py3-none-any.whl → 0.2.6.dev3__py3-none-any.whl

phasenet/utils/inference.py

@@ -85,7 +85,8 @@ def predict_2d(model: torch.nn.Module,
         max_batch_size: Max patches per forward pass (memory safety valve).
 
     Returns:
-        Phase scores of shape (3, nx, nt) — [noise, P, S].
+        Phase scores of shape (3, nx, nt) — [noise, P, S]; with a polarity head (4, nx, nt), the fourth row the
+        probability that the first motion is up.
         If model outputs event_center/event_time, returns
         (scores, event_center, event_time) at their native resolution.
     """
@@ -167,6 +168,10 @@ def predict_2d(model: torch.nn.Module,
             with torch.autocast(device_type=dev.type, dtype=torch.bfloat16, enabled=bf16):
                 out = model({"data": batch})
             probs = torch.softmax(out["phase"].float(), dim=1)
+            if "polarity" in out:  # P(up), carried as a fourth row through the same overlap-discard
+                probs = torch.cat([probs, torch.sigmoid(out["polarity"].float())], dim=1)
+            if scores_sum.shape[0] != probs.shape[1]:
+                scores_sum = torch.zeros(probs.shape[1], nx, nt, dtype=torch.float32, device=acc_device)
             if not use_gpu:
                 probs = probs.cpu()
 
@@ -328,6 +333,8 @@ def detect_picks(scores: np.ndarray,
                     "phase_score": round(prob, 4),
                     "phase_type":  phase,
                     "dt_s":        dt_s,
+                    **({"phase_polarity": round(scores[3, ch_idx, tidx].item(), 4)}
+                       if phase == "P" and scores.shape[0] > 3 else {}),  # P(up) at the P pick
                 })
 
     return picks

{phasenet-0.2.6.dev2.dist-info → phasenet-0.2.6.dev3.dist-info}/METADATA

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

{phasenet-0.2.6.dev2.dist-info → phasenet-0.2.6.dev3.dist-info}/RECORD

@@ -24,11 +24,11 @@ phasenet/models/prompt/prompt_encoder.py,sha256=q1m9N-3Ir_b6otiX23X5PNjAS45y0i8u
 phasenet/models/prompt/transformer.py,sha256=upHVeXMeVF0wKRif2hduo9zzY3RhllzMs2tq1DjINgA,8317
 phasenet/utils/__init__.py,sha256=evtQmqHe-22D1JB_3TMdwEIS48druYsCPm7A__txmmA,56
 phasenet/utils/detect_peaks_cpu.py,sha256=rfmCt7kmYoaD9628Gqg35AfHNJUSyaUbR2jYsW4MVS4,7776
-phasenet/utils/inference.py,sha256=y-IOZ8ca9xgo05kTyh0GAtgO1z2pDHZt23sH2pVkkoQ,19978
+phasenet/utils/inference.py,sha256=CmTO4qOaY74HrH0Cdo45SNw3vqMQAQ_-ELuHzqPU6P8,20610
 phasenet/utils/postprocess.py,sha256=WhVsJuf7goASeW0HSTaR-dPUsbMJjXmFzMlXfNoJF3w,11749
 phasenet/utils/visualization.py,sha256=EHw1AMnOUPGQzBYGVpb33Ipc6jM3-EnnYdRfinpf6q0,27003
-phasenet-0.2.6.dev2.dist-info/METADATA,sha256=Vo4FEvtLYmUUrmIKrBiAK68jnmt4lU1x0f00tVcPY2o,23211
-phasenet-0.2.6.dev2.dist-info/WHEEL,sha256=mOdXR9C3wMVov7cSY8PY-XCzo3SmmWCHvGQr5csJoao,91
-phasenet-0.2.6.dev2.dist-info/entry_points.txt,sha256=m41YTuKlfR4svomlQKHVgq1_bLz4ng6gH8PbVBD1vqc,45
-phasenet-0.2.6.dev2.dist-info/top_level.txt,sha256=Zw-8M-227zxFhDWHIS13TKNYEz1cAGDhp7ADNunSgAI,21
-phasenet-0.2.6.dev2.dist-info/RECORD,,
+phasenet-0.2.6.dev3.dist-info/METADATA,sha256=C3kuZG_lmPH8gvFHGoLlnLe3s-5Nt8zyUqR_jksJ-9k,23211
+phasenet-0.2.6.dev3.dist-info/WHEEL,sha256=mOdXR9C3wMVov7cSY8PY-XCzo3SmmWCHvGQr5csJoao,91
+phasenet-0.2.6.dev3.dist-info/entry_points.txt,sha256=m41YTuKlfR4svomlQKHVgq1_bLz4ng6gH8PbVBD1vqc,45
+phasenet-0.2.6.dev3.dist-info/top_level.txt,sha256=Zw-8M-227zxFhDWHIS13TKNYEz1cAGDhp7ADNunSgAI,21
+phasenet-0.2.6.dev3.dist-info/RECORD,,