一种目标关键点检测方法、装置、电子设备及介质

AITNT-国内领先的一站式人工智能新闻资讯网站
# 热门搜索 #
一种目标关键点检测方法、装置、电子设备及介质
申请号:CN202411931057
申请日期:2024-12-26
公开号:CN119360008B
公开日期:2025-04-22
类型:发明专利
摘要
本申请实施例公开了一种目标关键点检测方法、装置、电子设备及介质。该方法包括:对目标关键点类型文本进行编码,得到关键点类型文本编码特征;基于样本图像对变换神经网络进行预测目标关键点训练的过程中,在解码器的自注意力层和交叉注意力层中加入关键点类型文本编码特征,得到对样本图像的目标关键点预测结果;根据目标关键点预测结果与标注信息对变换神经网络进行调整优化,得到目标关键点检测模型,基于目标关键点检测模型对待检测图像进行目标关键点检测。上述方案加入关键点类型文本编码特征,能够使模型学习到目标的相同部位相同类型关键点的通用特征,提高了目标关键点的检测准确率以及模型的收敛速度。
技术关键词
编码特征 查询特征 注意力 关键点检测方法 矩阵 文本 校正模块 解码器 样本 图像 前馈神经网络 坐标 层级 网格 电子设备 处理器 可读存储介质 通用特征