Эх сурвалжийг харах

Merge pull request #565 from FlyingQianMM/develop_qh

add `from osgeo import gdal` when use high-version gdal
Jason 4 жил өмнө
parent
commit
be583426ad

+ 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: