数据集笔记：Beijing-BRT-dataset

XMU-smartdsp/Beijing-BRT-dataset (github.com)

1 数据集介绍

• 这个数据集包含1,280张图片，标注了16,795名行人，用于人群分析。
• 使用720张图片进行训练，560张图片进行测试。
• 名为"frame"的文件夹包含人群图像。 名为"ground_truth"的文件夹包含真实标签。
  • 例如，'1-20170325134657.jpg' 对应于 '1-20170325134657.mat'
  • 此图像中第i个人的真实位置由每行的位置[x,y]给出。

2 数据集结构

3 数据集处理

以1-20160620091411 为例

图像如下

import scipy.io


data = scipy.io.loadmat('1-20160620091411.mat')
#读取.mat文件


data.keys()
#显示加载文件的键值

1. __header__：MAT文件的头部信息。
2. __version__：MAT文件的版本。
3. __globals__：全局变量。
4. loc：数据变量。

 

data['__header__']
#b'MATLAB 5.0 MAT-file, Platform: PCWIN64, Created on: Wed Jun 07 10:52:42 2017'

data['__version__']
#'1.0'

data['__globals__']
#[]


data['loc']
'''
array([[143. , 573.5],
       [ 56. , 490. ],
       [185.5, 274. ],
       [173.5, 139.5],
       [200.5,  95.5],
       [233.5,  81. ],
       [271. ,  61.5],
       [251.5,  61. ],
       [208. ,  79.5],
       [326. , 477.5],
       [259. , 404.5],
       [291.5, 106.5],
       [323. , 134.5],
       [326.5,  12. ]])
'''

第一行 [143.,573.5]表示一个位置，其中 x=143 且 y=573.5