把坐标系转换代码隔离出来

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")
+
+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)