rectangle不对 是pol啥的

如果你希望结果中的类型为多边形（Polygon）而是矩形（Rectangle），你可以修改代码如下：

import numpy as np
import cv2
import json

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()  # 将分割掩码转换为列表形式
        })

# 将结果转换为JSON字符串
json_results = json.dumps(results)

在上述代码中，我们使用`cv2.findContours`函数找到分割掩码的轮廓，并将轮廓坐标转换为列表形式。然后，将轮廓添加到结果字典中的`polygon`字段中。

请注意，这里假设`masks`是一个包含分割掩码的形状数组（n，W，H）。如果`masks`不是这种形式，请根据实际情况进行修改。最后，我们将结果转换为JSON字符串，以便进一步处理或存储。