coco数据集转换SAM2格式
coco是一个大json汇总了所有train的标签
SAM2训练一张图对应一个json标签
import json
import os
from pycocotools import mask as mask_utils
import numpy as np
import cv2
def poly2mask(points, width, height):
points_array = np.array(points, dtype=np.int32).reshape(-1, 2)
mask = np.zeros((height, width), dtype=np.uint8) # 注意顺序是(height, width)
cv2.fillPoly(mask, [points_array], 255) # 填充多边形区域为255
return mask2rle(mask)
def mask2rle(mask):
"""将二值化掩码转换为RLE编码"""
rle = mask_utils.encode
原文地址:https://blog.csdn.net/weixin_44298961/article/details/144379321
免责声明:本站文章内容转载自网络资源,如本站内容侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!