[发明专利]图片识别方法及装置

说明书

本发明提供了一种图片识别方法及装置，涉及家装设计的技术领域，所述方法包括：对获取的待识别图片进行预处理，并采用卷积神经网络提取待识别图片包括的目标产品的低频特征，采用跳跃连接的方式提取待识别图片中的高频细节特征；采用卷积神经网络确定目标产品的类别，根据目标产品的特征对目标产品采用YOLOv3回归方法与锚点机制对目标产品进行定位，并在目标产品所在的位置生成目标边框；关联显示目标边框和目标产品的类别。本发明能够对家居产品模型图及家居贴图进行分类、识别与定位，识别结果与所分类别准确率高，能够为家居设计的方案优化和家居产品推荐提供保障。

技术领域

本发明涉及家装设计技术领域，尤其是涉及一种图片识别方法及装置。

背景技术

目前，在家装设计领域中，伴随AI(Artificial Intelligence，人工智能)智能化应用在家居产品设计中的不断深入，对家居产品模型图或家居贴图的自动分类、定位与识别工作变得极为重要。在实际场景与渲染场景中，例如现在流行的装修方案效果图当中，针对只有图片信息的情形下，只有先找到目标产品(如图片中的沙发、茶几)的位置并确定其类别，才能进行方案的设计优化与修改，以及进行下一步的包括产品推荐的工作。另外，对众多家居产品的商家而言，在自身图片数据库及设计方案的管理过程中，都是通过人工对家居产品模型图与贴图进行自动分类与识别工作的，工作效率低下。因此，目前缺少对家居产品模型图或贴图的自动分类、定位与识别的方案。

发明内容

本发明的目的在于提供一种图片识别方法及装置，能够对家居产品模型图及家居贴图进行分类、识别与定位，识别结果与所分类别准确率高，能够为家居设计的方案优化和家居产品推荐提供保障。

第一方面，本发明提供的一种图片识别方法，包括：

获取待识别图片，所述待识别图片包括目标产品；

提取所述待识别图片包括的所述目标产品的特征，并根据所述目标产品的特征确定所述目标产品的类别；

在所述待识别图片中对所述目标产品进行定位，并在所述目标产品所在的位置生成目标边框；

关联显示所述目标边框和所述目标产品的类别。

第二方面，本发明提供的一种图片识别装置，包括获取模块、特征提取模块、定位模块和显示模块：

所述获取模块用于获取待识别图片，所述待识别图片包括目标产品；

所述特征提取模块用于提取所述待识别图片包括的所述目标产品的特征，并根据所述目标产品的特征确定所述目标产品的类别；

所述定位模块用于在所述待识别图片中对所述目标产品进行定位，并在所述目标产品所在的位置生成目标边框；

所述显示模块用于关联显示所述目标边框和所述目标产品的类别。

第三方面，本发明提供了一种电子设备，包括存储器、处理器及存储在所述存储器上并可在所述处理器上运行的计算机程序，所述处理器执行所述计算机程序以实现第一方面所述的方法。

第四方面，本发明提供了一种具有处理器可执行的非易失的程序代码的计算机可读介质，所述程序代码使所述处理器执行第一方面所述的方法。

本发明提供的图片识别方法和装置，通过提取待识别图片上目标产品的特征，并根据提取的特征对目标产品进行分类识别；然后对目标产品进行定位，生成目标边框；最后，将目标产品的类别和位置进行显示；本发明能够对家居产品模型图及家居贴图进行识别与分类，识别结果与所分类别准确率高，能够为家居设计的方案优化和家居产品推荐提供保障，大大降低了家居设计的工作量，提高了工作效率。

附图说明