Просмотр исходного кода

add when use high-version gdal

FlyingQianMM 4 лет назад
Родитель
Сommit
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: