在Python中的Loadind NYU深度v2数据集

时间:2018-06-04 06:35:10

标签: python kinect mat depth

我试图验证深度算法,因此我需要一些好的数据。

我想使用let str = "35x35"; let pattern = /^\d+x\d+$/i; console.log(str.match(pattern)[0]);http://cs.nyu.edu/~silberman/datasets/nyu_depth_v2.html)。

我有标记的数据集,但文件格式为'.mat'。

使用h5py时,我得到的彩色图像值是

NYU Depth V2 dataset

是否有可能将颜色文件保存在png文件中或使用实际值加载它们,而无需在python中进行缩放?

def image_loader():

[[ 255 255 255],[255 255 255] ,[255 255 255]...

1 个答案:

答案 0 :(得分:0)

我遇到了同样的问题,当我尝试将其加载到PIL中时,颜色正好位于图像的中间。我发现了一些使用ndarray.transpose的解决方案,使形状从(3,640,480)变为(480,640,3)。这样我就可以正确查看图像了。

Image.fromarray(f['images'][0].transpose(2,1,0))