results = [] if masks.shape[0] > 0: for label, score, seg in zip(class_ids, conf, masks): # 解析分割掩码 mask = seg.argmax(axis=0) # 取最大值的索引，将二维掩码还原为一维 mask = mask.astype(np.uint8) # 将掩码转换为无符号整数类型 mask_binary = np.where(mask > 0, 255, 0) # 将掩码转换为二值图像 # 获取分割多边形 contours, _ = cv2.findContours(mask_binary, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE) contour = contours[0] if len(contours) > 0 else np.array([]) contour = contour.flatten().tolist() # 将多边形坐标转换为列表形式 results.append({ "confidence": str(score), "label": labels.get(label, "unknown"), "polygon": contour, "type": "polygon", "mask": mask.tolist() # 将分割掩码转换为列表形式 })怎么改

你可以将结果列表 `results` 的初始化放在 `if` 语句的外面，并将 `results` 作为一个参数传递给你的函数，以便在函数中修改它。这样，即使不存在分割掩码，`results` 也会保持为空列表。

下面是修改后的代码示例：

results = []

if masks.shape[0] > 0:
    for label, score, seg in zip(class_ids, conf, masks):
        # 解析分割掩码
        mask = seg.argmax(axis=0)  # 取最大值的索引，将二维掩码还原为一维
        mask = mask.astype(np.uint8)  # 将掩码转换为无符号整数类型
        mask_binary = np.where(mask > 0, 255, 0)  # 将掩码转换为二值图像

        # 获取分割多边形
        contours, _ = cv2.findContours(mask_binary, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
        contour = contours[0] if len(contours) > 0 else np.array([])
        contour = contour.reshape(-1, 2).tolist()  # 将多边形坐标转换为列表形式

        results.append({
            "confidence": str(score),
            "label": labels.get(label, "unknown"),
            "polygon": contour,
            "type": "polygon",
            "mask": mask.tolist()  # 将分割掩码转换为列表形式
        })

在这个示例中，我们在 `if` 语句外部初始化了 `results` 列表，并将其作为参数传递给你的函数。同时，我们还对多边形的坐标进行了一些修改，使用 `reshape(-1, 2)` 将多边形坐标重新组织成每行两个元素的形式，然后转换为列表形式。

请注意，这只是一个示例，你需要根据你的具体需求对代码进行适当的修改。