[Fix] Fix unit test image file path error (#1835)

This commit is contained in:
MengzhangLI 2022-07-28 16:30:29 +08:00 committed by GitHub
parent 1ded0a4278
commit 3bbdd6dc4a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -97,13 +97,13 @@ class TestSegLocalVisualizer(TestCase):
image = mmcv.imread( image = mmcv.imread(
osp.join( osp.join(
osp.dirname(__file__), osp.dirname(__file__),
'../data/pseudo_cityscapes_dataset/leftImg8bit/frankfurt_000000_000294_leftImg8bit.png' # noqa '../data/pseudo_cityscapes_dataset/leftImg8bit/val/frankfurt/frankfurt_000000_000294_leftImg8bit.png' # noqa
), ),
'color') 'color')
sem_seg = mmcv.imread( sem_seg = mmcv.imread(
osp.join( osp.join(
osp.dirname(__file__), osp.dirname(__file__),
'../data/pseudo_cityscapes_dataset/gtFine/frankfurt_000000_000294_gtFine_labelTrainIds.png' # noqa '../data/pseudo_cityscapes_dataset/gtFine/val/frankfurt/frankfurt_000000_000294_gtFine_labelTrainIds.png' # noqa
), ),
'unchanged') 'unchanged')
sem_seg = torch.unsqueeze(torch.from_numpy(sem_seg), 0) sem_seg = torch.unsqueeze(torch.from_numpy(sem_seg), 0)