[Fix] Fix DATASETS not imported error (#1827)

This commit is contained in:
MengzhangLI 2022-07-26 14:46:59 +08:00 committed by GitHub
parent 3cf2bbda2a
commit 6758b148b3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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