diff --git a/modelscope/outputs.py b/modelscope/outputs.py
index d80ba9c5..92e3410b 100644
--- a/modelscope/outputs.py
+++ b/modelscope/outputs.py
@@ -153,7 +153,7 @@ TASK_OUTPUTS = {
 
     # semantic segmentation result for single sample
     #   {
-    #       "masks": [np.array # 2D array containing only 0, 255]
+    #       "masks": [np.array # 2D array with shape [height, width]]
     #   }
     Tasks.semantic_segmentation: [OutputKeys.MASKS],