Как загрузить .png и .npy из ссылок в тензор потока - PullRequest
Новая
       19

Как загрузить .png и .npy из ссылок в тензор потока

0 голосов
/ 04 ноября 2019

Я изучаю тензор потока. В качестве личного проекта я хочу обучить модель оценивать расстояние от изображения RGB в каждом пикселе. Я хочу использовать набор данных DIODE для обучения модели. У меня есть три переменные, первая содержит пути к * .png изображениям, вторая содержит пути к * .npy файлам, представляющим глубину, а третья содержит пути к * .npy файлам, представляющим маску достоверностидатчик. Я использую этот код для сортировки файлов в папке. 

base_dir = val_data
def getListOfFiles(dirName, extension):
  allFiles=list()
  for sub in os.listdir(dirName):

    fullPath = os.path.join(dirName, sub)
    if os.path.isdir(fullPath):
      allFiles = allFiles + getListOfFiles(fullPath, extension)
    else:
      if fullPath.endswith(extension):
        allFiles.append(fullPath)
  return allFiles

rgbFiles = getListOfFiles(base_dir, '.png')
depthFiles = getListOfFiles(base_dir, 'depth.npy')
maskFiles = getListOfFiles(base_dir, 'mask.npy')
rgbFiles.sort()
depthFiles.sort()
maskFiles.sort()

Теперь я хочу загрузить изображения и соответствующий файл глубины, который будет использоваться в методе подгонки.

Возможно ли этопомочь мне сделать это?

Спасибо

...