『图像分割笔记』代码积累

---

一、标签可视化

标签图像的值只包含 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 图像不准确。