diff --git a/kelp_o_matic/cli.py b/kelp_o_matic/cli.py index 168e0cd..c02b7e8 100644 --- a/kelp_o_matic/cli.py +++ b/kelp_o_matic/cli.py @@ -44,7 +44,7 @@ def find_kelp( Detect kelp in image at path SOURCE and output the resulting classification raster to file at path DEST. """ - find_kelp_(source, dest, species, crop_size, use_nir, tuple(band_order), use_gpu) + find_kelp_(source, dest, species, crop_size, use_nir, band_order, use_gpu) @cli.command() @@ -55,6 +55,12 @@ def find_mussels( 1024, help="The data window size to run through the segmentation model.", ), + band_order: Optional[list[int]] = typer.Option( + None, + "-b", + help="GDAL-style band re-ordering flag. Defaults to RGB order. " + "To e.g., reorder a BGR image at runtime, pass flags `-b 3 -b 2 -b 1`.", + ), use_gpu: bool = typer.Option( True, "--gpu/--no-gpu", help="Enable or disable GPU, if available." ), @@ -63,7 +69,7 @@ def find_mussels( Detect mussels in image at path SOURCE and output the resulting classification raster to file at path DEST. """ - find_mussels_(source, dest, crop_size, use_gpu) + find_mussels_(source, dest, crop_size, band_order, use_gpu) def version_callback(value: bool) -> None: