如何从链接中加载.png和.npy（以十为单位）

Question

我正在学习tensorflow。 作为个人项目，我想训练一个模型来评估每个像素中与RGB图像的距离。 我想使用二极管数据集来训练模型。 我有三个变量，第一个包含到*.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()

现在我想加载图像和相应的深度文件，以便在fit方法中使用

有没有可能指导我这样做

谢谢