From 6dd3456b2866fb2eb8e4fe02a376d1e7b4bd7ba8 Mon Sep 17 00:00:00 2001 From: Guillaume-Duret <guillaume.duret@ec-lyon.fr> Date: Fri, 2 Jun 2023 15:12:48 +0200 Subject: [PATCH] gt seg --- datasets/linemod/dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/datasets/linemod/dataset.py b/datasets/linemod/dataset.py index 2582a40..f060ac3 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)) -- GitLab