NRCan / geo-deep-learning

Deep learning applied to georeferenced datasets
https://geo-deep-learning.readthedocs.io/en/latest/
MIT License
149 stars 49 forks source link

Add Tag with classes dict to inference raster #504

Closed CharlesAuthier closed 1 year ago

CharlesAuthier commented 1 year ago

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.