open-mmlab / mmsegmentation

OpenMMLab Semantic Segmentation Toolbox and Benchmark.
https://mmsegmentation.readthedocs.io/en/main/
Apache License 2.0
8.02k stars 2.58k forks source link

[Features] Add support for Kitti semantic segmentation dataset #1599

Open AkideLiu opened 2 years ago

AkideLiu commented 2 years ago

Describe the feature

Add support for KITTI semantic segmentation dataset

Motivation

Kitti semantic segmentation dataset is a lightweight dataset for semantic segmentation which shares the same label policy as cityscapes. It's an excellent starting point for segmentation and employs the weights pre-trained on cityscapes to perform transfer-learning, do you consider to support this dataset.

Related resources

http://www.cvlibs.net/datasets/kitti/eval_semseg.php?benchmark=semantics2015

Additional context

Happy to create a PR.

MeowZheng commented 2 years ago

Thanks in advance for your contribution to mmseg, and looking forward to your nice PR.

AkideLiu commented 2 years ago

Thanks in advance for your contribution to mmseg, and looking forward to your nice PR.

Thanks for your quick reply, I have the following questions looking forward to your reply.

As this dataset has 200 training images with the fine label, however, the test set is not provided, do I need to implement the train_test_split for this dataset. In my current using scenario, the training set has been split from the original training set with 150 images and 50 images for both testing and validation datasets.

if the train_test_split is required, could you please provide a guideline for the split ratio and reference code for insight?