diff --git a/post_pro/trans_orthophoto.py b/trans_orthophoto.py similarity index 71% rename from post_pro/trans_orthophoto.py rename to trans_orthophoto.py index 7d4297b..dc0ca61 100644 --- a/post_pro/trans_orthophoto.py +++ b/trans_orthophoto.py @@ -1,4 +1,5 @@ import logging +import argparse import rasterio from rasterio.warp import calculate_default_transform, reproject, Resampling @@ -6,6 +7,8 @@ from rasterio.warp import calculate_default_transform, reproject, Resampling class TransOrthophoto: def __init__(self): self.logger = logging.getLogger('UAV_Preprocess.TransOrthophoto') + logging.basicConfig(level=logging.INFO, + format='%(asctime)s %(levelname)s %(message)s') def trans_to_epsg4326(self, ori_orthophoto_path: str, output_path: str): # 打开原始文件 @@ -37,8 +40,13 @@ class TransOrthophoto: resampling=Resampling.nearest ) - self.logger.info(f"文件已成功重投影") + self.logger.info(f"文件已成功重投影: {output_path}") -if __name__ == "__main__": - trans = TransOrthophoto() - trans.trans_to_epsg4326(r"G:\ODM_output\test\orthophoto.tif", r"G:\ODM_output\test\orthophoto_epsg4326.tif") \ No newline at end of file + +parser = argparse.ArgumentParser(description="正射影像投影转换为EPSG:4326") +parser.add_argument('--input_img', required=True, help='输入影像路径') +parser.add_argument('--output_img', required=True, help='输出影像路径') +args = parser.parse_args() + +trans = TransOrthophoto() +trans.trans_to_epsg4326(args.input_img, args.output_img)