图像处理之颜色检测分类标记(Python OpenCV实现)

最近开始接触图像处理，接到的首个任务就是将实验室用颜色标记好的数据再在原图上按不同颜色框出来，以在模型预测阶段检查预测效果。下面使用一张摇滚乐队Halestrom的图片进行说明。

首先，我拿到的原图如下图所示：

图1

我们将原始图片按照人、地板、墙三种元素进行标记，得到下图：

图2

将上述两张图片输入我们的模型，那么模型能够做到给出一张新的图片它就能够输出一张按颜色分类标记元素的图片。我们这里要做的是，将按颜色分类标记元素的区块转换为线框绘制在原始图片上，便于检测预测效果，具体需要得到的图片如下图所示：

图3