Skip to content

Commit

Permalink
[Fix] fix wrong variables passing for set_dataset_meta (#3348)
Browse files Browse the repository at this point in the history
  • Loading branch information
likyoo committed Sep 26, 2023
1 parent 2b3d91a commit cf8fc7b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion mmseg/apis/mmseg_inferencer.py
Original file line number Diff line number Diff line change
Expand Up @@ -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],
Expand Down

0 comments on commit cf8fc7b

Please sign in to comment.