摘要
提供一种基于神经网络的图像目标模板提取方法,该方法包括:获取指定目标的图像并预处理,获得标准化的图像数据集;对图像数据集完成目标类型及模板区域标注,形成标注数据集;对图像数据集及标注数据集进行划分,得到训练集、验证集和测试集;基于训练集、验证集和测试集训练获得目标模板提取网络模型,用于根据输入目标类型,输出图像中符合输入目标类型的精确模板;其中,目标模板提取网络模型包括模板提取网络模型和类别判别网络模型,模板提取网络模型用于生成多个模板区域,类别判别网络模型用于判断每个模板区域的目标类型;将所需目标类型及未知图像集输入目标模板提取网络模型,提取得到目标模板。有效提高了目标模板生成的鲁棒性。
技术关键词
模板提取方法
网络
sigmoid函数
融合图像特征
训练集
图像编码器
水面舰艇
数据
可读存储介质
解码器
处理器
电子设备
模块
语义
飞机
存储器
鲁棒性