Adding a tag containing the classes_dict parameter, find in the dataset yaml, to the output raster after inference.
Also adding a tests to verify the output, important to use the fetch_tag_raster function to fetch the classes_dict, otherwise the output will be a string and not a dictionary.
Adding a tag containing the
classes_dict
parameter, find in the dataset yaml, to the output raster after inference. Also adding a tests to verify the output, important to use thefetch_tag_raster
function to fetch theclasses_dict
, otherwise the output will be a string and not a dictionary.