CLI for inference using fastai models
Remove all transformations so predict_segmentation works with any image size and returns same sized output than input.
DataLoader.set_base_transforms
DataLoader.set_base_transforms ()
Removes all transforms with a size parameter
Pipeline.remove
Pipeline.remove (t)
Remove an instance of t from self if present
predict_segmentation
predict_segmentation (path_to_model:str, path_to_image:str, outfile:str, processing_dir:str='temp', tile_size:int=400, tile_overlap:int=100, use_tta:bool=False)
Segment image into land cover classes with a pretrained models TODO save also information about label and class TODO add test-time augmentations
predict_segmentation_fastai
predict_segmentation_fastai (path_to_model:str, path_to_image:str, outfile:str, processing_dir:str='temp', tile_size:int=400, tile_overlap:int=100, use_tta:bool=False)
CLI for semantic segmentation with fastai
| Type | Default | Details | |
|---|---|---|---|
| path_to_model | str | Path to pretrained model file | |
| path_to_image | str | Path to image to annotate | |
| outfile | str | Path and filename for output raster | |
| processing_dir | str | temp | Directory to save the intermediate tiles. Deleted after use |
| tile_size | int | 400 | Tile size to use. Default 400x400px tiles |
| tile_overlap | int | 100 | Tile overlap to use. Default 100px |
| use_tta | bool | False | Use test-time augmentation |