From 27b9d4412e2df4d49d2ab5765c49d964b094c9ed Mon Sep 17 00:00:00 2001 From: Hunor Laczko Date: Tue, 20 Sep 2022 22:54:28 +0000 Subject: [PATCH] small dataset fixes --- dataset.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/dataset.py b/dataset.py index d5dbf8b..c5a952b 100644 --- a/dataset.py +++ b/dataset.py @@ -79,7 +79,7 @@ class StaticDataset(): if 'uv_static' + res_string in self.signature.keys(): uv_static = np.load(os.path.join(self.folder, 'uv_static' + self.filled + res_string, id + '.npz')); uv_static = uv_static['uv_static' + self.filled]; uv_static = np.transpose(uv_static, [2, 0, 1]); data['uv_static' + res_string] = uv_static - if 'uv_mask' in self.signature.keys(): + if 'uv_mask' + res_string in self.signature.keys(): uv_mask = np.load(os.path.join(self.folder, 'uv_mask' + res_string, id + '.npz')); uv_mask=uv_mask['uv_mask']; uv_mask = uv_mask.astype(np.float32); data['uv_mask' + res_string] = uv_mask @@ -211,7 +211,7 @@ class DatasetFactory(): if 'uv_curvature' + res_string in signature.keys(): uv_curvature = np.load(os.path.join(folder, 'uv_curvature' + filled + res_string, id + '_' + frame_nr + '.npz')); - if filled != '': + if filled != '' or res_string != '': uv_curvature = uv_curvature['uv_curvature' + filled] else: uv_curvature = uv_curvature['curvature_uv']