def _get_training_queue(self, batch_size):
image_batch, depth_batch, self.depth_bins = edof_reader.get_edof_training_queue(
opt.img_dir,
patch_size=self.patch_size,
batch_size=batch_size,
num_depths=3,
color=True)
return (image_batch, depth_batch), image_batch
def _get_training_queue(self, batch_size):
image_batch, _, _ = edof_reader.get_edof_training_queue(opt.img_dir,
patch_size=self.patch_size,
batch_size=batch_size)
return image_batch, image_batch