def _read(self, files):
     if any(['Depth' in f for f in files]):
         #returns depth in meters
         images = np.asarray([open_depth_synthia(f) for f in files])
     else:
         images = np.asarray([open_rgb(f) for f in files])
     return images
Example #2
0
 def _read(self, files):
     from utils.utils import open_rgb, open_depth_freiburg
     if any(['pfm' in f for f in files]):
         images = np.asarray([open_depth_freiburg(f) for f in files])
     else:
         images = np.asarray([open_rgb(f) for f in files])
     return images
Example #3
0
 def _read(self, files):
     from utils.utils import open_depth_synthia, open_rgb, open_depth
     if any(['depth' in f for f in files]):
         images = np.asarray([open_depth(f) for f in files])[...,
                                                             np.newaxis]
     else:
         images = np.asarray([open_rgb(f) for f in files])
     return images
Example #4
0
 def _read(self, files):
     from utils.utils import open_depth_tum, open_rgb
     if any(['depth' in f for f in files]):
         #returns depth in meters
         images = np.asarray([open_depth_tum(f) for f in files])
     else:
         images = np.asarray([open_rgb(f) for f in files])
     return images
Example #5
0
    def _read(self, files):
        from utils.utils import open_depth_mp, open_rgb
        if any(['color' in f for f in files]):

            images = np.asarray([open_rgb(f) for f in files])
        else:
            #returns depth
            images = np.asarray([open_depth_mp(f) for f in files])
        return images