Skip to content

Commit

Permalink
update for json keys
Browse files Browse the repository at this point in the history
  • Loading branch information
gferraro committed Apr 24, 2024
1 parent 7c1e545 commit 10609ec
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions Melt/identify_tracks.py
Original file line number Diff line number Diff line change
Expand Up @@ -325,19 +325,18 @@ def classify(file, models,analyse_tracks,meta_data = None):
length = get_end(frames, sr)
signals = signal_noise(frames[: int(sr * length)], sr, 281)
# want to use signals for chrips

if analyse_tracks:
if meta_data is None:
return
meta_tracks = [t for t in meta_data["Tracks"] if t.get("classify")]
meta_tracks = [t for t in meta_data["Tracks"] ]
tracks = []
for t in meta_tracks:
pos = t["data"]["positions"][0]
pos = t["positions"][0]
freq_start = pos["y"] * MAX_FRQUENCY
freq_end = pos["height"]*MAX_FRQUENCY + freq_start

# add to signals also???
signal = Signal(t["data"]["start_s"], t["data"]["end_s"], freq_start,freq_end)
signal = Signal(t["start"], t["end"], freq_start,freq_end)
signal.track_id = t["id"]
tracks.append(signal)
else:
Expand Down

0 comments on commit 10609ec

Please sign in to comment.