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