along/UAV
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

z轴平移

Browse Source
This commit is contained in:
weixin_46229132 2025-02-20 20:04:14 +08:00
parent 3b6af50acc
commit 696fe7fe7e
1 changed files with 24 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
post_pro/conv_obj.py
View File

@ -34,20 +34,25 @@ class ConvertOBJ:
first_grid_dir, "odm_orthophoto", "odm_orthophoto_log.txt") first_grid_dir, "odm_orthophoto", "odm_orthophoto_log.txt")
self.ref_east, self.ref_north = self.read_utm_offset(log_file) 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(): for grid_id in grid_points.keys():
try: try:
self._convert_single_grid(grid_id, grid_points) self._convert_single_grid(grid_id, grid_points)
except Exception as e: except Exception as e:
self.logger.error(f"网格 {grid_id} 转换失败: {str(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): def _convert_single_grid(self, grid_id, grid_points):
"""转换单个网格的OBJ文件""" """转换单个网格的OBJ文件"""
# 1. 构建相关路径 # 1. 构建相关路径
grid_name = f"grid_{grid_id[0]}_{grid_id[1]}" grid_name = f"grid_{grid_id[0]}_{grid_id[1]}"
project_dir = os.path.join(self.output_dir, grid_name, "project") 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") texturing_dst_dir = os.path.join(project_dir, "odm_texturing_dst")
opensfm_dir = os.path.join(project_dir, "opensfm") opensfm_dir = os.path.join(project_dir, "opensfm")
log_file = os.path.join( log_file = os.path.join(
@ -83,7 +88,7 @@ class ConvertOBJ:
tile_dir, f"Tile_{grid_id[0]}_{grid_id[1]}.osgb") tile_dir, f"Tile_{grid_id[0]}_{grid_id[1]}.osgb")
shutil.copy2(output_osgb, target_osgb) shutil.copy2(output_osgb, target_osgb)
def _create_merged_metadata(self): def _create_merged_metadata(self, high):
"""创建合并后的metadata.xml文件""" """创建合并后的metadata.xml文件"""
# 转换为WGS84经纬度 # 转换为WGS84经纬度
center_lon, center_lat = self.transformer.transform( center_lon, center_lat = self.transformer.transform(
@ -91,7 +96,7 @@ class ConvertOBJ:
metadata_content = f"""<?xml version="1.0" encoding="utf-8"?> metadata_content = f"""<?xml version="1.0" encoding="utf-8"?>
<ModelMetadata version="1"> <ModelMetadata version="1">
<SRS>EPSG:4326</SRS> <SRS>EPSG:4326</SRS>
<SRSOrigin>{center_lon},{center_lat},0.000000</SRSOrigin> <SRSOrigin>{center_lon},{center_lat},{high}</SRSOrigin>
<Texture> <Texture>
<ColorSource>Visible</ColorSource> <ColorSource>Visible</ColorSource>
</Texture> </Texture>
@ -211,3 +216,18 @@ class ConvertOBJ:
img = cv2.imread(dst_path, cv2.IMREAD_UNCHANGED) img = cv2.imread(dst_path, cv2.IMREAD_UNCHANGED)
self.logger.info( self.logger.info(
f"下采样后文件大小: {file_size / (1024 * 1024):.2f} MB") 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