mirror of
https://github.com/open-mmlab/mmsegmentation.git
synced 2025-06-03 22:03:48 +08:00
[Fix] Fix DATASETS not imported error (#1827)
This commit is contained in:
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…
x
Reference in New Issue
Block a user