diff --git a/mmseg/apis/mmseg_inferencer.py b/mmseg/apis/mmseg_inferencer.py index 0775d2afd8..095639a80f 100644 --- a/mmseg/apis/mmseg_inferencer.py +++ b/mmseg/apis/mmseg_inferencer.py @@ -84,7 +84,7 @@ def __init__(self, self.model = revert_sync_batchnorm(self.model) assert isinstance(self.visualizer, SegLocalVisualizer) - self.visualizer.set_dataset_meta(palette, classes, dataset_name) + self.visualizer.set_dataset_meta(classes, palette, dataset_name) def _load_weights_to_model(self, model: nn.Module, checkpoint: Optional[dict],