diff --git a/.gitignore b/.gitignore index 4eec394..53c5944 100644 --- a/.gitignore +++ b/.gitignore @@ -171,4 +171,11 @@ test_data/ logs/ wandb/ *.pth -checkpoints/ \ No newline at end of file +checkpoints/ + +data/gt_mesh +data/gt_point +data/step +data/input_data +data/output_data +data/name_list.txt \ No newline at end of file diff --git a/brep2sdf/config/default_config.py b/brep2sdf/config/default_config.py index 372f2eb..96e71bf 100644 --- a/brep2sdf/config/default_config.py +++ b/brep2sdf/config/default_config.py @@ -35,26 +35,10 @@ class DataConfig: # pre_process origin_brep_dir: str = '/mnt/mynewdisk/dataset/furniture/step/furniture_dataset_step/' - # 数据路径 - pkl_path = "/home/wch/brep2sdf/test_data/pkl/train/bathtub_0004.pkl" - sdf_path = "/home/wch/brep2sdf/test_data/sdf/train/bathtub_0004.npz" - # 基础路径配置 base_data_dir: str = '/home/wch/brep2sdf/test_data' # 基础数据目录 - @property - def brep_dir(self) -> str: - return os.path.join(self.base_data_dir, 'pkl') - - @property - def sdf_dir(self) -> str: - return os.path.join(self.base_data_dir, 'sdf') - - @property - def valid_data_dir(self) -> str: - return os.path.join(self.base_data_dir, 'result/pkl') - @dataclass @@ -84,10 +68,9 @@ class TrainConfig: # 保存路径 checkpoint_dir: str = '/home/wch/brep2sdf/checkpoints' # 结果保存目录 - # 文件命名 - model_name: str = 'brep2sdf' # 模型名称,用于文件命名 - checkpoint_format: str = '{model_name}_epoch_{epoch:03d}.pth' # 检查点文件名格式 - best_model_name: str = '{model_name}_best.pth' # 最佳模型文件名格式 + # 文件命名 # 模型名称,用于文件命名 + + best_model_name: str = '{model_name}/best.pth' # 最佳模型文件名格式 @dataclass