diff --git a/post_pro/conv_obj.py b/post_pro/conv_obj.py index b7af753..764786e 100644 --- a/post_pro/conv_obj.py +++ b/post_pro/conv_obj.py @@ -34,20 +34,25 @@ class ConvertOBJ: first_grid_dir, "odm_orthophoto", "odm_orthophoto_log.txt") self.ref_east, self.ref_north = self.read_utm_offset(log_file) + # 以第一个grid计算高度的偏移量 + min_z = self.get_min_z_from_obj(os.path.join(first_grid_dir, 'odm_texturing', 'odm_textured_model_geo.obj')) + min_25d_z = self.get_min_z_from_obj(os.path.join(first_grid_dir, 'odm_texturing_25d', 'odm_textured_model_geo.obj')) + high = min_25d_z - min_z + for grid_id in grid_points.keys(): try: self._convert_single_grid(grid_id, grid_points) except Exception as e: self.logger.error(f"网格 {grid_id} 转换失败: {str(e)}") - self._create_merged_metadata() + self._create_merged_metadata(high) def _convert_single_grid(self, grid_id, grid_points): """转换单个网格的OBJ文件""" # 1. 构建相关路径 grid_name = f"grid_{grid_id[0]}_{grid_id[1]}" project_dir = os.path.join(self.output_dir, grid_name, "project") - texturing_dir = os.path.join(project_dir, "odm_texturing_25d") + texturing_dir = os.path.join(project_dir, "odm_texturing") texturing_dst_dir = os.path.join(project_dir, "odm_texturing_dst") opensfm_dir = os.path.join(project_dir, "opensfm") log_file = os.path.join( @@ -83,7 +88,7 @@ class ConvertOBJ: tile_dir, f"Tile_{grid_id[0]}_{grid_id[1]}.osgb") shutil.copy2(output_osgb, target_osgb) - def _create_merged_metadata(self): + def _create_merged_metadata(self, high): """创建合并后的metadata.xml文件""" # 转换为WGS84经纬度 center_lon, center_lat = self.transformer.transform( @@ -91,7 +96,7 @@ class ConvertOBJ: metadata_content = f""" EPSG:4326 - {center_lon},{center_lat},0.000000 + {center_lon},{center_lat},{high} Visible @@ -211,3 +216,18 @@ class ConvertOBJ: img = cv2.imread(dst_path, cv2.IMREAD_UNCHANGED) self.logger.info( f"下采样后文件大小: {file_size / (1024 * 1024):.2f} MB") + + def get_min_z_from_obj(self, file_path): + min_z = float('inf') # 初始值设为无穷大 + with open(file_path, 'r') as obj_file: + for line in obj_file: + # 检查每一行是否是顶点定义(以 'v ' 开头) + if line.startswith('v '): + # 获取顶点坐标 + parts = line.split() + # 将z值转换为浮动数字 + z = float(parts[3]) + # 更新最小z值 + if z < min_z: + min_z = z + return min_z