神经网络训练的方法、电子设备及计算机存储介质

基本信息

申请号 CN202111307529.4 申请日 -
公开(公告)号 CN114170438A 公开(公告)日 2022-03-11
申请公布号 CN114170438A 申请公布日 2022-03-11
分类号 G06V10/46(2022.01)I;G06V10/80(2022.01)I;G06V10/82(2022.01)I;G06N3/04(2006.01)I;G06N3/08(2006.01)I;G06T7/55(2017.01)I 分类 计算;推算;计数;
发明人 常青玲;崔岩;杨鑫;廖洹浩;王煜涵 申请(专利权)人 中德(珠海)人工智能研究院有限公司
代理机构 广州嘉权专利商标事务所有限公司 代理人 梁国平
地址 529000广东省江门市蓬江区东成村22号
法律状态 -

摘要

摘要 本发明公开了一种神经网络训练的方法、电子设备及计算机存储介质,包括如下步骤:获取预设的训练样本图;获取预测特征图;基于预设的边界感知深度损失函数,以预设的训练样本图为输入,预测特征图为输出,训练神经网络模型,得到训练后的神经网络模型。本发明通过设置边界感知深度损失函数,能够有效提高神经网络模型在训练过程中对边界区域的关注度,保证边界区域的深度和深度梯度具备良好的正确性,从而有效抑制平滑现象,避免出现边界区域模糊的问题,最终能够有效提高边界预测精度。