TSP6K
收藏TSP6K 数据集概述
数据集简介
TSP6K 数据集是一个用于交通场景解析的数据集,相关论文和代码可在 arXiv 上找到。该数据集的代码实现基于开源的语义分割工具箱 MMsegmentation。
数据集准备
数据集可从 此链接 下载,并放置在 /data/TSP6K/ 目录下。由于论文尚未被接受,下载数据集前需发送电子邮件至 pt.jiang at mail.nankai.edu.cn 获取密码。
数据集结构如下:
data ├── TSP6K │ ├── image │ ├── label │ ├── split
训练
使用以下命令训练 SegNext 模型,结合提出的 Detail Refining Decoder: bash bash tools/dist_train.sh configs/tsp6k/segnext_base_1024x1024_160k_tsp6k_msaspp_rrm_5tokens_12heads.py 8 --auto-resume
评估
结果和模型
| 方法 | 主干网络 | 裁剪尺寸 | 学习率计划 | val mIoU (ms) | val iIoU (ms) | 配置文件 | 模型 |
|---|---|---|---|---|---|---|---|
| SegNext+DRD | MSCAN-B | 1024x1024 | 160000 | 75.8 | 58.4 | config | model |
| SegNext+DRD | MSCAN-L | 1024x1024 | 160000 | 76.2 | 58.9 | config | model |
提供预训练的分割模型,可通过以下命令直接评估: bash bash tools/dist_test.sh configs/tsp6k/segnext_base_1024x1024_160k_tsp6k_msaspp_rrm_5tokens_12heads.py ./work_dirs/segnext_base_1024x1024_160k_tsp6k_msaspp_rrm_5tokens_12heads/latest.pth 8 --out ./work_dirs/segnext_base_1024x1024_160k_tsp6k_msaspp_rrm_5tokens_12heads/results.pkl --aug-test --eval mIoU
使用 iIoU 指标评估分割模型: bash bash tools/dist_test.sh configs/tsp6k/segnext_base_1024x1024_160k_tsp6k_msaspp_rrm_5tokens_12heads.py ./work_dirs/segnext_base_1024x1024_160k_tsp6k_msaspp_rrm_5tokens_12heads/latest.pth 8 --out ./work_dirs/segnext_base_1024x1024_160k_tsp6k_msaspp_rrm_5tokens_12heads/results.pkl --aug-test --eval cityscapes
引用
如果 TSP6K 数据集和分割网络对您的研究有用,请引用:
@inproceedings{jiang2024traffic, title={Traffic Scene Parsing through the TSP6K Dataset}, author={Jiang, Peng-Tao and Yang, Yuqi and Cao, Yang and Hou, Qibin and Cheng, Ming-Ming and Shen, Chunhua}, booktitle={Proceedings of the IEEE/CVF conference on computer vision and pattern recognition}, year={2024} }




