lionel/batdetect2
1
0
Fork 0
mirror of https://github.com/macaodha/batdetect2.git synced 2026-04-04 15:20:19 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add audio dir to data cli command

Browse Source
This commit is contained in:
mbsantiago 2026-03-28 11:09:47 +00:00
parent 5d92f3a00d
commit 9ec2f1a107
1 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
src/batdetect2/cli/data.py
View File

@ -80,11 +80,17 @@ def summary(
type=click.Path(exists=True), type=click.Path(exists=True),
help="The base directory to which all recording and annotations paths are relative to.", help="The base directory to which all recording and annotations paths are relative to.",
) )
@click.option(
"--audio-dir",
type=click.Path(exists=True),
help="The directory containing the audio files. All paths will be relative to this directory.",
)
def convert( def convert(
dataset_config: Path, dataset_config: Path,
field: str | None = None, field: str | None = None,
output: Path = Path("annotations.json"), output: Path = Path("annotations.json"),
base_dir: Path | None = None, base_dir: Path | None = None,
audio_dir: Path | None = None,
): ):
"""Convert a dataset config file to soundevent format.""" """Convert a dataset config file to soundevent format."""
from soundevent import data, io from soundevent import data, io
@ -103,4 +109,12 @@ def convert(
description=config.description, description=config.description,
) )
io.save(annotation_set, output) if audio_dir:
audio_dir = Path(audio_dir)
if not audio_dir.is_absolute():
audio_dir = audio_dir.resolve()
print(f"Using audio directory: {audio_dir}")
io.save(annotation_set, output, audio_dir=audio_dir)