基于改进YoloV4网络的虹膜定位算法

作者：杨亚男; 朱晓冬; 刘元宁^*; **; 董霖

来源：吉林大学学报(理学版), 2022, 60(02): 369-380.

DOI：10.13413/j.cnki.jdxblxb.2021202

• 虹膜定位
• YoloV4
• 快速soft-DIoU-NMS算法
• K-means++聚类

针对传统虹膜定位算法很难完成准确定位导致识别效果不稳定的问题，提出一种基于改进YoloV4网络的虹膜定位算法.首先利用YoloV4结合MobileNetV3对虹膜内外圆进行粗定位，再利用瞳孔、虹膜和巩膜的灰度差值分别对虹膜内外圆进行精定位.同时，使用K-means++聚类算法生成先验框；使用快速soft-DIoU-NMS算法去除预测过程冗余框，提高算法检测率；使用Focal Loss作为类别损失函数.对比实验结果表明，该算法运行速度更快，定位准确率更高，识别算法的效果更明显.

• 单位
  吉林大学