TSP6K
收藏资源简介:
TSP6K数据集是一个专门为交通监控场景设计的,包含来自交通监控场景的图像,具有高质量的像素级和实例级标注。该数据集捕捉了比现有驾驶场景更为拥挤的交通场景,交通参与者数量是现有驾驶场景的数倍。
The TSP6K dataset is specifically tailored for traffic monitoring scenarios. It comprises images captured from traffic monitoring environments, with high-quality pixel-level and instance-level annotations. This dataset captures traffic scenes that are more congested than existing driving scenarios, and the number of traffic participants is several times that of such existing scenarios.
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} }




