lionel/batdetect2
1
0
Fork 0
mirror of https://github.com/macaodha/batdetect2.git synced 2026-05-22 22:32:18 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

docs: clarify train base-dir option

Browse Source
This commit is contained in:
mbsantiago 2026-04-30 16:51:24 +01:00
parent 9da05c172c
commit f82ec218f0
1 changed files with 17 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
src/batdetect2/cli/train.py
View File

@ -24,6 +24,14 @@ __all__ = ["train_command"]
"training starts from a fresh model config." "training starts from a fresh model config."
), ),
) )
@click.option(
"--base-dir",
type=click.Path(exists=True),
help=(
"Base directory used to resolve relative paths inside the training "
"and validation dataset configs."
),
)
@click.option( @click.option(
"--targets", "--targets",
"targets_config", "targets_config",
@ -111,6 +119,7 @@ def train_command(
model_path: Path | None = None, model_path: Path | None = None,
ckpt_dir: Path | None = None, ckpt_dir: Path | None = None,
log_dir: Path | None = None, log_dir: Path | None = None,
base_dir: Path | None = None,
targets_config: Path | None = None, targets_config: Path | None = None,
model_config: Path | None = None, model_config: Path | None = None,
training_config: Path | None = None, training_config: Path | None = None,
@ -191,7 +200,10 @@ def train_command(
model_conf = model_conf.model_copy(update={"targets": target_conf}) model_conf = model_conf.model_copy(update={"targets": target_conf})
logger.info("Loading training dataset...") logger.info("Loading training dataset...")
train_annotations = load_dataset_from_config(train_dataset) train_annotations = load_dataset_from_config(
train_dataset,
base_dir=base_dir,
)
logger.debug( logger.debug(
"Loaded {num_annotations} training examples", "Loaded {num_annotations} training examples",
num_annotations=len(train_annotations), num_annotations=len(train_annotations),
@ -199,7 +211,10 @@ def train_command(
val_annotations = None val_annotations = None
if val_dataset is not None: if val_dataset is not None:
val_annotations = load_dataset_from_config(val_dataset) val_annotations = load_dataset_from_config(
val_dataset,
base_dir=base_dir,
)
logger.debug( logger.debug(
"Loaded {num_annotations} validation examples", "Loaded {num_annotations} validation examples",
num_annotations=len(val_annotations), num_annotations=len(val_annotations),