[Fix]: complement necessary argument of seg_suffix of cityscapes (#9330)

Issue: #9329
A new entry `seg_suffix` is added for CustomDataset.__init__()
but missing in `self_coco = CocoDataset` in function `evaluate()`,
`mmdet/datasets/cityscapes.py`.

Signed-off-by: Shengjiang QUAN <qsj287068067@126.com>

Signed-off-by: Shengjiang QUAN <qsj287068067@126.com>
pull/9435/head
AdorableJiang 3 years ago committed by GitHub
parent 671213116e
commit 91f47c5ca0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 5
      mmdet/datasets/cityscapes.py

@ -263,8 +263,9 @@ class CityscapesDataset(CocoDataset):
# create CocoDataset with CityscapesDataset annotation
self_coco = CocoDataset(self.ann_file, self.pipeline.transforms,
None, self.data_root, self.img_prefix,
self.seg_prefix, self.proposal_file,
self.test_mode, self.filter_empty_gt)
self.seg_prefix, self.seg_suffix,
self.proposal_file, self.test_mode,
self.filter_empty_gt)
# TODO: remove this in the future
# reload annotations of correct class
self_coco.CLASSES = self.CLASSES

Loading…
Cancel
Save