将图像裁剪成单独的部分

Question

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

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

Answer 1

你可以这样做：

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