diff --git a/fastdeploy/vision/evaluation/__init__.py b/fastdeploy/vision/evaluation/__init__.py
index d2c9a7911..f7dfadd1c 100644
--- a/fastdeploy/vision/evaluation/__init__.py
+++ b/fastdeploy/vision/evaluation/__init__.py
@@ -14,3 +14,4 @@
 from __future__ import absolute_import
 from .classify import eval_classify
 from .detection import eval_detection
+from .segmentation import eval_segmentation
diff --git a/fastdeploy/vision/evaluation/segmentation.py b/fastdeploy/vision/evaluation/segmentation.py
index 7e0d21e1a..85de6b661 100644
--- a/fastdeploy/vision/evaluation/segmentation.py
+++ b/fastdeploy/vision/evaluation/segmentation.py
@@ -22,8 +22,8 @@ import time
 
 def eval_segmentation(model, data_dir):
     import cv2
-    from utils import Cityscapes
-    from utils import f1_score, calculate_area, mean_iou, accuracy, kappa
+    from .utils import Cityscapes
+    from .utils import f1_score, calculate_area, mean_iou, accuracy, kappa
     assert os.path.isdir(
         data_dir), "The image_file_path:{} is not a directory.".format(
             data_dir)
@@ -76,20 +76,3 @@ def eval_segmentation(model, data_dir):
             average_inference_time
         ]))
     return eval_metrics
-
-
-import fastdeploy as fd
-#model = fd.vision.segmentation.PaddleSegModel("/huangjianhui/temp/FastDeploy/model_zoo/vision/ppseg/unet/unet_Cityscapes/model.pdmodel",
-#                              "/huangjianhui/temp/FastDeploy/model_zoo/vision/ppseg/unet/unet_Cityscapes//model.pdiparams",
-#                              "/huangjianhui/temp/FastDeploy/model_zoo/vision/ppseg/unet/unet_Cityscapes/deploy.yaml")
-#
-option = fd.RuntimeOption()
-option.use_paddle_backend()
-option.use_gpu(3)
-model = fd.vision.segmentation.PaddleSegModel(
-    "/huangjianhui/temp/FastDeploy/model_zoo/vision/ppseg/PP-LiteSeg/output_no_static_size/model.pdmodel",
-    "/huangjianhui/temp/FastDeploy/model_zoo/vision/ppseg/PP-LiteSeg/output_no_static_size/model.pdiparams",
-    "/huangjianhui/temp/FastDeploy/model_zoo/vision/ppseg/PP-LiteSeg/output_no_static_size/deploy.yaml",
-    option)
-
-result = eval_segmentation(model, "/huangjianhui/PaddleSeg/data/cityscapes/")