lionel/batdetect2
1
0
Fork 0
mirror of https://github.com/macaodha/batdetect2.git synced 2025-06-29 22:51:58 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

Better frequency axis tick values

Browse Source
This commit is contained in:
Santiago Martinez 2023-04-07 15:46:04 -06:00
parent 0a7ad18193
commit 85092e846b
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
batdetect2/plot.py
View File

@ -5,6 +5,7 @@ from typing import List, Optional, Tuple, Union, cast
import numpy as np import numpy as np
import torch import torch
from matplotlib import axes, patches from matplotlib import axes, patches
import matplotlib.ticker as tick
from matplotlib import pyplot as plt from matplotlib import pyplot as plt
from batdetect2.detector.parameters import DEFAULT_PROCESSING_CONFIGURATIONS from batdetect2.detector.parameters import DEFAULT_PROCESSING_CONFIGURATIONS
@ -91,7 +92,12 @@ def spectrogram(
ax.imshow(spec, aspect="auto", origin="lower", cmap=cmap, extent=extent) ax.imshow(spec, aspect="auto", origin="lower", cmap=cmap, extent=extent)
ax.set_xlabel("Time (s)") ax.set_xlabel("Time (s)")
ax.set_ylabel("Frequency (Hz)") ax.set_ylabel("Frequency (kHz)")
def y_fmt(x, _):
return f"{x / 1000:d}"
ax.yaxis.set_major_formatter(tick.FuncFormatter(y_fmt))
return ax return ax