Skip to content

Main

Source

Module for CLI functionality and entrypoint.

Contains the CLI entrypoint, the run function; and first level subcommands train and predict. The conf subcommand is added through the app.add_typer function, and its implementation is contained in the conf.py file.

predict(model, source, batch_size=1, tile_size=None, tile_overlap=(48, 48, -1), axes=None, data_type='tiff', tta_transforms=False, write_type='tiff', work_dir=None, prediction_dir=Path('predictions'))

Create and save predictions from CAREamics models.

run()

CLI Entry point.

train(source, train_source, train_target=None, val_source=None, val_target=None, use_in_memory=True, val_percentage=0.1, val_minimum_split=1, work_dir=None)

Train CAREamics models.