license:apache-2.0 size_categories:1K<n<10K format:parquet modality:image library:datasets library:dask library:mlcroissant library:polars region:us
---
license: apache-2.0
dataset_info:
  features:
  - name: image
    dtype: image
  - name: label
    dtype: image
  splits:
  - name: train
    num_bytes: 2160695876.134
    num_examples: 2999
  - name: validation
    num_bytes: 909490805.613
    num_examples: 1207
  download_size: 2934142939
  dataset_size: 3070186681.7469997
---
# TSP6K-dataset
This dataset is from the official repository. URL: https://github.com/PengtaoJiang/TSP6K

You would need `id2label` variable to assign pixel values to class names.
```python
import json

id2label = {"0": "road", "1": "sidewalk", "2": "building",
            "3": "wall", "4": "railing", "5": "vegetation", 
            "6": "terrain", "7": "sky", "8": "person", 
            "9": "rider", "10": "car", "11": "truck",
            "12": "bus", "13": "motorcycle", "14": "bicycle", 
            "15": "indication line", "16": "lane line", "17": "crosswalk", 
            "18": "pole", "19": "traffic light", "20": "traffic sign"
            }

with open('id2label.json', 'w') as fp:
  json.dump(id2label, fp)
```