Browse Source

use pyrockos argmax

beam_shifter
Marius Kriegerowski 4 years ago
parent
commit
aa567793bf
  1. 4
      src/core.py

4
src/core.py

@ -7,7 +7,7 @@ import shutil
import numpy as num
from pyrocko import pile, trace, util, io
from pyrocko.parstack import parstack
from pyrocko.parstack import parstack, argmax as pargmax
from lassie import common, plot, grid as gridmod
@ -232,7 +232,7 @@ def scan(
config.detector_threshold, shift_max + 1.0/fsmooth_min)
tr_stackmax_indx = tr_stackmax.copy(data=False)
imaxs = num.argmax(frames, axis=0)
imaxs = pargmax(frames)
tr_stackmax_indx.set_ydata(imaxs.astype(num.int32))
tr_stackmax_indx.set_location('i')

Loading…
Cancel
Save