声明:基于python3.6提取NYUD V2数据集
NYUD V2数据集下载网址如下:
https://cs.nyu.edu/~silberman/datasets/nyu_depth_v2.html(文中在Toolbox中提供了常用的预处理图像函数)
下载好的数据集的文件名是nyu_depth_v2_labeled.mat
,可以用matlab打开。
NYUD V2数据集由1449张640*480的图像组成。
提取原图:
import numpy as np
import matplotlib.pyplot as plt
import scipy.io as sio
import h5py
import os
f=h5py.File("/home/yuyangyg/DataSet/NYUDataV2/nyu_depth_v2_labeled.mat")
images=f["images"]
images=np.array(images)
path_converted='./nyu_images'
if not os.path.isdir(path_converted):
os.makedirs(path_converted)
from PIL import Image
images_number=[]
for i in range(len(images)):
print(i)
images_number.append(images[i])
a=np.array(images_number[i])
# print len(img)
&nb