VPGNet数据集解析

VPGNet交通标志数据集

VPGNet是一个包含了四个场景(晴天，夜间，阴雨等）下的交通路况数据集，其主要是针对地面标志（箭头线，人行道）的任务，该数据集一共包括以下17类地面标注

0. 背景 (background)
1. 实线白色车道线 (lane_solid_white)
2. 断续白色车道线 (lane_broken_white)
3. 双实线白色车道线 (lane_double_white)
4. 实线黄色车道线 (lane_solid_yellow)
5. 断续黄色车道线 (lane_broken_yellow)
6. 双实线黄色车道线 (lane_double_yellow)
7. 断续蓝色车道线 (lane_broken_blue)
8. 缓慢行驶车道线 (lane_slow)
9. 停车线 (stop_line)
10. 左转箭头 (arrow_left)
11. 右转箭头 (arrow_right)
12. 直行箭头 (arrow_go_straight)
13. 掉头箭头 (arrow_u_turn)
14. 减速带 (speed_bump)
15. 人行横道 (crossWalk)
16. 安全区域 (safety_zone)
17. 其他道路标线 (other_road_markings)

标注信息的数量

数据下载

数据集下载

该数据集的所有有用信息均保存在一个(640×480×5)的mat文件中。
数据集的关键信息均保存在(640x480x5)的数组中，前三个通道为图片的rgb三通道信息，第四维通道为图像的17类二值分割图像(像素值为0-17)，第五维通道为该论文提出的vanish point信息

Vanishing point（消失的点）

消失点标注：【EASY：一个清晰的场景（如笔直的道路）；HARD：混乱的场景（如交通堵塞）；NONE：消失点不存在的地方（如十字路口）】。直线和曲线车道都被用来预测消失点

数据解析

将mat解析保存为如下格式

VPGNet_new
├── images
│   ├── 1.jpg
│   ├── 10.jpg
│   ├── ...
├── labels
│   ├── 1.png
│   ├── 10.png
│   ├── ...
└── points
│   ├── 1.npy
│   ├── 10.npy
│   ├── ...

解析代码

import os
import numpy as np
import scipy.io
import imageio

# Define the paths
path = 'VPGNet'
savedir_img =