一种基于深度学习的单阶段服装颜色识别系统及方法

摘要

本公开提供了一种基于深度学习的单阶段服装颜色识别系统及方法，旨在解决现有技术识别速度较低的问题。系统包括：服装分割模型分支对服装图像做服装分割得到服装区域图像；颜色分类模型分支对服装图像进行RGB颜色识别，得到像素的RGB值根据像素的RGB值将服装图像进行像素归类，将像素归类为对应的颜色簇；融合单元根据服装图像的像素归类与服装区域图像，得到服装区域图像的像素归类；根据服装区域图像的颜色簇内像素的RGB值，得到服装区域图像中每个颜色簇的颜色表示；映射匹配单元将服装区域图像颜色簇的颜色表示在Lab空间内和目标颜色进行距离度量，将距离最近的目标颜色作为服装区域图像颜色簇的颜色类别。本公开的系统识别速度快且运算量较小。