一种自然场景图像中的服装定位与分割方法

针对自然场景图像中背景复杂、目标多的问题,提出一种基于深度学习和图像处理的服装定位与分割方法。首先,基于YOLOv4深度学习算法训练人体目标检测网络模型,根据模型定位并分割出图像中包含服装的人物;接着,采用卷积姿态机（convolutional pose machines, CPM）训练上衣、外套、连衣裙、半身裙及裤子五类服装关键点网络模型;然后,对包含服装的人物图像进行归一化处理,选择服装类型,再根据上述服装关键点网络模型输出服装的关键点;最后,根据这些关键点信息,形成初始矩阵框,使用OneCut算法将服装图像分割出来。实验证明,在复杂背景和多目标的自然图像中,文章方法具有误差小、准确率高、运算速度快的特点,能够满足自然场景图像中服装自动分割的要求。