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: