From 5cc47de234c608f6f912255084793090af55cab0 Mon Sep 17 00:00:00 2001 From: ioanalzr Date: Tue, 27 May 2025 12:23:23 +0100 Subject: [PATCH] custom training labels --- beneuro_pose_estimation/cli.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/beneuro_pose_estimation/cli.py b/beneuro_pose_estimation/cli.py index 3a5880d..8c6d89b 100644 --- a/beneuro_pose_estimation/cli.py +++ b/beneuro_pose_estimation/cli.py @@ -223,15 +223,19 @@ def train( "If not provided, uses default cameras from params.default_cameras." ), ), + custom_labels: bool = typer.Option( + False, + "--custom-labels", "-cl", + help="If set, use custom labels when training the models.", + ), ): """ Train SLEAP models for specified cameras (or all defaults). """ from beneuro_pose_estimation.sleap.sleapTools import train_models - cams = cameras or params.default_cameras - # train_models is the function you already wrote - train_models(cam, custom_labels = custom_labels) + cams = cameras or params.default_cameras + train_models(cameras=cams, custom_labels=custom_labels) # =================================== Updating ==========================================