将图像裁剪成单独的部分

Question

我使用语义分割对下图所示图像中的不同元素进行颜色编码。

在Python中，我想根据第二张图的颜色把原图裁剪成很多小图，沙发就变成一个裁剪的部分，lamp就变成了one.etc。沙发上枕头的重叠可以忽略不计。假设我有一个图像的 3D 数组，我想将该数组分成单独的彩色部分，并在裁剪原始图像时应用这些元素的坐标。我该如何实现？

Answer 1

你可以这样做：

• 找出分割图像中唯一颜色的数量 - 请参阅
• 遍历该颜色列表，使该颜色为白色，其他颜色为黑色，然后 findContours() 获取边界框并将该边界框的内容保存为 PNG。