-
Notifications
You must be signed in to change notification settings - Fork 303
Conversation
Can you resolve conflicts? |
Rebased to master |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Can you add doc link to docs/source/dataset.rst
?
chainercv/datasets/__init__.py
Outdated
@@ -3,6 +3,8 @@ | |||
from chainercv.datasets.camvid.camvid_dataset import camvid_label_names # NOQA | |||
from chainercv.datasets.camvid.camvid_dataset import CamVidDataset # NOQA | |||
from chainercv.datasets.cityscapes.cityscapes_semantic_segmentation_dataset import CityscapesSemanticSegmentationDataset # NOQA | |||
from chainercv.datasets.cityscapes.cityscapes_test_image_dataset import CityscapesTestImageDataset # NOQA | |||
from chainercv.datasets.cityscapes.cityscapes_utils import cityscapes_labels # NOQA |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Can you delete this line? This may confuse users because there is not sufficient description about this object.
We are only exposing label_names
and label_colors
directly below chainercv.datasets
.
|
||
class CityscapesTestImageDataset(dataset.DatasetMixin): | ||
|
||
"""Dataset class for test images of `Cityscapes dataset`_. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Image dataset for test split of `Cityscapes dataset`_.
This is more consistent with other Image dataset
(e.g. ADE20KTestImageDataset
).
@yuyu2172 Fixed. |
Please add the dataset to |
oops, missed it. added. |
|
Sorry. I found that the test fails. |
Also, I think we should add |
@yuyu2172 Added teardown method. |
def setUp(self): | ||
self.temp_dir = tempfile.mkdtemp() | ||
img_dir = os.path.join( | ||
self.temp_dir, 'leftImg8bit/{}/aachen'.format(self.split)) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
self.split raises an error
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Sorry, fixed.
LGTM |
Please merge #437 first.
This PR adds
CityscapesTestImageDataset
class.