diff --git a/paddlex/cv/datasets/seg_dataset.py b/paddlex/cv/datasets/seg_dataset.py
index af1f9dfcaf9946459f9a4fdd693f4f0c2d129b44..7f770a19a916c1698cbee327a017876c18047d4d 100644
--- a/paddlex/cv/datasets/seg_dataset.py
+++ b/paddlex/cv/datasets/seg_dataset.py
@@ -14,6 +14,7 @@
 
 from __future__ import absolute_import
 import os.path as osp
+import platform
 import random
 import copy
 import paddlex.utils.logging as logging