From 29a79a77f32b33a0349d7d76bfa78c016f2350a7 Mon Sep 17 00:00:00 2001 From: Honghyun Ahn <32119804+hyunynim@users.noreply.github.com> Date: Tue, 20 Oct 2020 16:51:26 +0900 Subject: [PATCH] Modify the path to load the mesh Previous path is ~/data --- core/dataset/mesh_dataset/mesh_loader.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/core/dataset/mesh_dataset/mesh_loader.py b/core/dataset/mesh_dataset/mesh_loader.py index b7ec358..cac66a1 100644 --- a/core/dataset/mesh_dataset/mesh_loader.py +++ b/core/dataset/mesh_dataset/mesh_loader.py @@ -33,7 +33,7 @@ def __init__(self, data_dir, experiment_directory, split_file, checkpoint_num=20 instance_list_new = [] for instance in self.instance_list: - fname = os.path.join(os.path.expanduser('~/data'), 'NormalizationParameters', 'ShapeNetV2', self.class_name, '{}.npz'.format(instance)) + fname = os.path.join(data_dir, 'NormalizationParameters', 'ShapeNetV2', self.class_name, '{}.npz'.format(instance)) if os.path.exists(fname): instance_list_new.append(instance) else: