增加生成训练数据

shape_spectral.py

@@ -390,9 +390,11 @@ def process_images(full_bil_path, mask, outdir='None', debug="None"):
 
     return combined_data
 
-# # # 示例：批量处理指定路径下的光谱图像和掩膜文件
-# bil_path = r'D:\WQ\test\Traindata-05'
-# mask_path = r'D:\WQ\test\mask'
-# outdir = r"D:\WQ\test"  # 输出文件夹路径
-#
-# process_images(bil_path, mask_path, outdir)
+if __name__ == "__main__":
+
+    # # 示例：批量处理指定路径下的光谱图像和掩膜文件
+    bil_path = r'D:\WQ\test\Traindata-05'
+    mask_path = r'D:\WQ\test\mask'
+    outdir = r"D:\WQ\test"  # 输出文件夹路径
+
+    process_images(bil_path, mask_path, outdir)