From 26e3984f76dec4ff76c54ec5698112fbde0adc76 Mon Sep 17 00:00:00 2001 From: mckay Date: Wed, 9 Apr 2025 14:38:44 +0800 Subject: [PATCH] =?UTF-8?q?=E5=8A=A0=E5=BC=BAgen=20name=20list=E5=8A=9F?= =?UTF-8?q?=E8=83=BD?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- data/scripts/gen_name_list.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/data/scripts/gen_name_list.py b/data/scripts/gen_name_list.py index 459c4b5..1dd5921 100644 --- a/data/scripts/gen_name_list.py +++ b/data/scripts/gen_name_list.py @@ -12,10 +12,11 @@ import os 保存id到 ../name_list.txt ''' -def get_valid_ids(): +def get_valid_ids(check_pt_exists=False): # 定义路径 - mesh_dir = "../gt_mesh_" + mesh_dir = "../gt_mesh" point_dir = "../gt_point" + pt_dir = "../output_data" # 获取所有可能的ID obj_files = [f for f in os.listdir(mesh_dir) if f.endswith('.obj')] @@ -28,14 +29,14 @@ def get_valid_ids(): has_obj = os.path.exists(os.path.join(mesh_dir, f"{id}.obj")) has_xyz = os.path.exists(os.path.join(point_dir, f"{id}_50k.xyz")) has_ptangle = os.path.exists(os.path.join(point_dir, f"{id}.ptangle")) - - if has_obj and has_xyz and has_ptangle: + has_pt = not check_pt_exists or os.path.exists(os.path.join(pt_dir, f"{id}.pt")) + if has_obj and has_xyz and has_ptangle and has_pt: valid_ids.append(id) return valid_ids if __name__ == "__main__": - valid_ids = get_valid_ids() + valid_ids = get_valid_ids(check_pt_exists=True) print(f"找到 {len(valid_ids)} 个有效ID:") # 保存到文件