浏览代码

add when use high-version gdal

FlyingQianMM 4 年之前
父节点
当前提交
97140a478f
共有 1 个文件被更改,包括 7 次插入4 次删除
  1. 7 4
      paddlex/cv/transforms/seg_transforms.py

+ 7 - 4
paddlex/cv/transforms/seg_transforms.py

@@ -72,10 +72,13 @@ class Compose(SegTransform):
             try:
                 import gdal
             except:
-                six.reraise(*sys.exc_info())
-                raise Exception(
-                    "Please refer to https://github.com/PaddlePaddle/PaddleX/tree/develop/examples/multi-channel_remote_sensing/README.md to install gdal"
-                )
+                try:
+                    from osgeo import gdal
+                except:
+                    raise Exception(
+                        "Please refer to https://github.com/PaddlePaddle/PaddleX/tree/develop/examples/multi-channel_remote_sensing/README.md to install gdal"
+                    )
+                    six.reraise(*sys.exc_info())
 
             dataset = gdal.Open(img_path)
             if dataset == None: