只是需要使用lableme标准的json文件获取mask,经过分析使用其他库的必要性比较低,于是自己写了一个通过labelme标准的json文件获取mask的程序。这里我使用的json文件只是标注了个polygon。
import cv2
import numpy as np
import json
json_path = r'D:\datasets\24\2022-01-05\10_color_1.json'
img_path = r'D:\datasets\24\2022-01-05\10_color_1.png'
labelme_json = json.load(open(json_path, encoding='utf-8'))
img = cv2.imread(img_path)
mask = np.zeros((720, 1280, 1), dtype=np.uint8)
mask2 = np.zeros((720, 1280, 3), dtype=np.uint8)
points = labelme_json['shapes'][0]['points']
points = np.array(points)
points = points.reshape(-1, 1, 2)
points = points.astype(np.int32)
cv2.fillConvexPoly(mask, points, (255,))
cv2.fillConvexPoly(mask2, points, (255,255,255))
# 显示mask
cv2.imshow('mask', mask)
cv2.waitKey(0)
# new_image = cv2.bitwise_and(img, mask)
new_image = cv2.bitwise_and(img, mask2)
# new_image[mask2==255] = 100
# 显示mask区域图像
cv2.imshow('new_image', new_image)
cv2.waitKey(0)
运行结果: