Loadind NYU глубина v2 набор данных в Python - PullRequest
Новая
       40

Loadind NYU глубина v2 набор данных в Python

0 голосов
/ 04 июня 2018

Я попытался проверить алгоритм глубины, и поэтому мне нужны хорошие данные.

Я хочу использовать NYU Depth V2 dataset (http://cs.nyu.edu/~silberman/datasets/nyu_depth_v2.html).

У меня есть помеченный набор данных, но файл в формате .mat.

При использовании h5py,значения цветных изображений, которые я получаю: 

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

Есть ли возможность сохранить файлы цветов в png-файле или загрузить их с реальными значениями, не масштабируя их в python?

def image_loader (): 

# data path
path_to_depth = './nyu_depth_v2_labeled.mat'

# read mat file
f = h5py.File(path_to_depth)

batch_size=6

pred = np.zeros((6,480,640,3))
gt = np.zeros((6,480,640,1))    
for i in range(batch_size):

    # read 0-th image. original format is [3 x 640 x 480], uint8
    img = f['images'][i]

    # reshape
    img_ = np.empty([480, 640, 3])
    img_[:,:,0] = img[0,:,:].T
    img_[:,:,1] = img[1,:,:].T
    img_[:,:,2] = img[2,:,:].T

    # read corresponding depth (aligned to the image, in-painted) of size [640 x 480], float64
    depth = f['depths'][i]

    depth_ = np.empty([480, 640])
    depth_[:,:] = depth[:,:].T

    pred[i,:,:,:] = img_
    gt[i,:,:,0] = depth_ 

return pred, gt
...