文章目录
一、标签可视化
标签图像的值只包含 0 和 1,所以它看起来整个图都是黑色的。
标签图像是 8 位的单通道图像,所以该方法支持最多 256 种类别。
如果直接打开图像的 Label 查看的话,应该是一个全黑的图像,因为对应的像素值太小,无法看清。如果想查看某个类别的标注情况,可以使用以下代码进行 label 标注情况的查看:
for shape in shapes:
categroy = shape["label"]
points = shape["points"]
points_array = np.array(points, dtype=np.int32)
if categroy == "Tom":
# 调试时将 Tom 这一类的填充颜色改为255,便于查看
mask = cv2.fillPoly(mask, [points_array], 255)
else:
mask = cv2.fillPoly(mask, [points_array], categroy_types.index(categroy))
cv2.imshow("mask", mask)
cv2,waitKey(0)
Note: 制作标签的时候要记得将其改回对应的索引值。最后,保存图像的时候要注意保存成 png
格式的图像,因为 jpg
格式会在存储时对图像进行压缩,导致 mask 图像不准确。