diff --git a/datasets/linemod/dataset.py b/datasets/linemod/dataset.py index 2582a401e3dbbd3ac08b07dbac7aaa0637fda8bf..f060ac310cee16a0ed6b39a0306659d57f484e12 100755 --- a/datasets/linemod/dataset.py +++ b/datasets/linemod/dataset.py @@ -61,7 +61,7 @@ class PoseDataset(data.Dataset): self.list_rgb.append('{0}/data/{1}/rgb/{2}.png'.format(self.root, '%02d' % item, input_line)) self.list_depth.append('{0}/data/{1}/depth/{2}.png'.format(self.root, '%02d' % item, input_line)) if self.mode == 'eval': - self.list_label.append('{0}/segnet_results/{1}_label/{2}.png'.format(self.root, '%02d' % item, input_line)) + self.list_label.append('{0}/segnet_results_gt/{1}_label/{2}.png'.format(self.root, '%02d' % item, input_line)) else: self.list_label.append('{0}/data/{1}/mask/{2}.png'.format(self.root, '%02d' % item, input_line))