图像识别方法、装置、电子设备及存储介质
摘要:
本发明涉及人工智能及数字医疗技术领域,提供一种图像识别方法、装置、电子设备及存储介质,所述方法包括:基于获取的训练图像集构造支持图像集和查询图像集;获取支持图像集中每个类别的原型特征;计算查询图像集中每个图像与每个类别的原型特征之间的距离;基于所述距离对原型网络的损失函数进行改进得到改进后的损失函数;根据改进后的损失函数对所述原型网络进行迭代训练,得到图像分类器。本发明在分类正确的基础上拉近相同类别的图像之间的距离且同时拉大了不同类别的图像之间的距离,使原型网络更加注意类别间和类别内的异同,从而训练出一个足够好的能够区分出相似和不同类别的嵌入空间,减少误分类率,进一步提高了模型的准确率。
0/0