[Fix] Fix DATASETS not imported error (#1827)
parent
3cf2bbda2a
commit
6758b148b3
|
@ -6,10 +6,10 @@ from unittest.mock import MagicMock
|
|||
|
||||
import pytest
|
||||
|
||||
from mmseg.datasets import (ADE20KDataset, BaseSegDataset,
|
||||
CityscapesDataset, COCOStuffDataset, ISPRSDataset,
|
||||
LoveDADataset, PascalVOCDataset, PotsdamDataset,
|
||||
iSAIDDataset)
|
||||
from mmseg.datasets import (ADE20KDataset, BaseSegDataset, CityscapesDataset,
|
||||
COCOStuffDataset, ISPRSDataset, LoveDADataset,
|
||||
PascalVOCDataset, PotsdamDataset, iSAIDDataset)
|
||||
from mmseg.registry import DATASETS
|
||||
from mmseg.utils import get_classes, get_palette
|
||||
|
||||
|
||||
|
|
Loading…
Reference in New Issue