def get_pred_objects(self, idx):
assert (idx < self.num_samples) #and self.split=='training'
pred_filename = os.path.join(self.pred_dir, '%06d.txt' % (idx))
is_exist = os.path.exists(pred_filename)
if is_exist:
return kitti_utils.read_label(pred_filename)
else:
return None
def get_label_objects(self, idx):
assert (idx < self.num_samples and self.split == 'training')
label_filename = os.path.join(self.label_dir, '%06d.txt' % (idx))
return utils.read_label(label_filename)
def get_label(self, idx):
label_file = os.path.join(self.label_path, '%06d.txt' % idx)
assert os.path.exists(label_file)
return kitti_utils.read_label(label_file)
def get_label_objects(self, idx):
if idx > self.num_samples:
return None
label_filename = os.path.join(self.label_dir, "%06d.txt" % idx)
return read_label(label_filename)