Open spcerrudo opened 5 months ago
Sorry, I forgot to consider the need for a color palette when designing the inference for large-scale remote sensing images. Additionally, known issues include displaying the progress of inference. These are expected to be addressed in the next version.
I have recently trained a model for remote sensing segmentation mapping. Unfortunately, the segment map displays the pixel classes in greyscale, not in the palette that I have set during the creation of custom dataset. The classes that are represented in the train model shows up in metrics / results so that is not the issue. Thank you very much for your time.
@DATASETS.register_module() class OpenEarthMapDataset(BaseSegDataset): """OpenEarthMap dataset.