wendi.hwd yingda.chen 3 years ago
parent
commit
939a9f2322
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      modelscope/outputs.py

+ 1
- 1
modelscope/outputs.py View File

@@ -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],



Loading…
Cancel
Save