NYU Depvh V2数据处理报错：TypeError: only integer scalar arrays can be converted to a scalar index

背景：复现BTS论文的时候，需要将NYU的数据集从mat转为image，执行下面代码。

代码链接：GitHub - cleinc/bts: From Big to Small: Multi-Scale Local Planar Guidance for Monocular Depth Estimation

在最后一行执行的时候会报错：

定位到代码88行：

添加87行代码，查看元素类型，发现输出的是 object ，也就是说明是一个引用，而不是基本的数值类型（如整数或浮点数）。

所以，我将第88行代码进行修改，视为引用去做处理，代码如下：

# 假设 sceneTypes 是一组引用
scene_refs = h5_file['sceneTypes'][0]
scenes = []
for ref in scene_refs:
    # 解引用
    obj = h5_file[ref]
    scene = ''.join(chr(c[0]) for c in obj)
    scenes.append(scene)

成功完成转换~