一种眼底相机定位的方法和装置与流程

本技术涉及眼底相机，特别涉及一种眼底相机定位的方法和装置。

背景技术：

1、眼底相机是一种用于拍摄眼睛内部，特别是视网膜和视神经盘的医疗成像设备。它对于诊断和监测多种眼病，如糖尿病视网膜病变、高血压视网膜病变、黄斑变性等具有重要作用。眼底相机工作过程中，电机需要把眼底相机镜头移动到相对于瞳孔中心点的特定位置。为此，需要对瞳孔中心点进行识别。

2、现有的识别瞳孔中心点的方法在瞳孔识别过程中，会损失瞳孔直径的精度，导致识别出的瞳孔中心点不准确，进而降低眼底相机拍摄眼底图像的成功率。

技术实现思路

1、为此，本技术公开如下技术方案：

2、本技术第一方面提供一种眼底相机定位的方法，包括：

3、获得目标对象的目标眼睛图像，所述目标眼睛图像至少包括所述目标对象的目标瞳孔；

4、根据预先获得的瞳孔识别模型处理所述目标眼睛图像，以确定所述目标眼睛图像中所述目标瞳孔的瞳孔中心点坐标和瞳孔尺寸，所述瞳孔识别模型包括用于处理所述目标眼睛图像的通道注意力模块和空间注意力模块；

5、根据所述瞳孔中心点坐标和所述瞳孔尺寸，控制眼底相机移动至所述瞳孔中心点坐标对应的目标位置。

6、可选的，获得所述瞳孔识别模型的方式包括：

7、加载包含预训练权重参数的初始识别模型；

8、根据所述初始识别模型处理样本眼睛图像，得到所述样本眼睛图像对应的标注文档，所述标注文档表征所述样本眼睛图像中瞳孔中心点坐标和瞳孔长宽；

9、根据所述标注文档和所述标注文档对应的修正后标注文档调整所述初始识别模型所含权重参数，直至获得模型损失满足收敛条件的识别模型；

10、在调整权重参数而获得的多个识别模型中，根据模型评估指标选择一个目标识别模型作为所述瞳孔识别模型。

11、可选的，获得所述样本眼睛图像的方式包括：

12、拍摄得到多帧初始眼睛图像；

13、对所述初始眼睛图像进行数据增强处理，获得多帧所述样本眼睛图像，所述数据增强处理包括缩放、旋转、翻转、模糊、混合中至少一种。

14、可选的，所述瞳孔识别模型包括目标检测网络，通道注意力模块和空间注意力模块；

15、所述根据预先获得的瞳孔识别模型处理所述目标眼睛图像，以确定所述目标眼睛图像中所述目标瞳孔的瞳孔中心点坐标，包括：

16、根据所述目标检测网络提取所述目标眼睛图像的第一图像特征；

17、根据所述通道注意力模块处理所述第一图像特征，得到所述目标眼睛图像的融合通道注意力特征；

18、根据所述空间注意力模块处理所述融合通道注意力特征，得到所述目标眼睛图像的融合空间注意力特征；

19、根据所述目标检测网络处理所述融合空间注意力特征，得到所述目标眼睛图像中所述目标瞳孔的瞳孔中心点坐标和瞳孔尺寸。

20、可选的，所述目标眼睛图像包括所述眼底相机拍摄得到的两帧目标眼睛图像，所述眼底相机为双目同步眼底相机；

21、所述根据所述瞳孔中心点坐标和所述瞳孔尺寸，控制眼底相机移动至所述瞳孔中心点坐标对应的目标位置，包括：

22、根据两帧所述目标眼睛图像的所述瞳孔中心点坐标和所述瞳孔尺寸，确定所述瞳孔中心点坐标对应的目标位置；

23、根据所述目标位置确定所述眼底相机的移动方向和移动距离；

24、根据所述移动方向和所述移动距离控制步进电机，使所述眼底相机的中心点被所述步进电机驱动而移动至所述目标位置。

25、本技术第二方面提供一种眼底相机定位的装置，包括：

26、获得单元，用于获得目标对象的目标眼睛图像，所述目标眼睛图像至少包括所述目标对象的目标瞳孔；

27、处理单元，用于根据预先获得的瞳孔识别模型处理所述目标眼睛图像，以确定所述目标眼睛图像中所述目标瞳孔的瞳孔中心点坐标和瞳孔尺寸，所述瞳孔识别模型包括用于处理所述目标眼睛图像的通道注意力模块和空间注意力模块；

28、控制单元，用于根据所述瞳孔中心点坐标和所述瞳孔尺寸，控制眼底相机移动至所述瞳孔中心点坐标对应的目标位置。

29、可选的，所述装置包括训练单元，用于按如下方式获得所述瞳孔识别模型的方式：

30、加载包含预训练权重参数的初始识别模型；

31、根据所述初始识别模型处理样本眼睛图像，得到所述样本眼睛图像对应的标注文档，所述标注文档表征所述样本眼睛图像中瞳孔中心点坐标和瞳孔长宽；

32、根据所述标注文档和所述标注文档对应的修正后标注文档调整所述初始识别模型所含权重参数，直至获得模型损失满足收敛条件的识别模型；

33、在调整权重参数而获得的多个识别模型中，根据模型评估指标选择一个目标识别模型作为所述瞳孔识别模型。

34、可选的，所述训练单元获得所述样本眼睛图像的方式包括：

35、拍摄得到多帧初始眼睛图像；

36、对所述初始眼睛图像进行数据增强处理，获得多帧所述样本眼睛图像，所述数据增强处理包括缩放、旋转、翻转、模糊、混合中至少一种。

37、可选的，所述瞳孔识别模型包括目标检测网络，通道注意力模块和空间注意力模块；

38、所述处理单元根据预先获得的瞳孔识别模型处理所述目标眼睛图像，以确定所述目标眼睛图像中所述目标瞳孔的瞳孔中心点坐标时，具体用于：

39、根据所述目标检测网络提取所述目标眼睛图像的第一图像特征；

40、根据所述通道注意力模块处理所述第一图像特征，得到所述目标眼睛图像的融合通道注意力特征；

41、根据所述空间注意力模块处理所述融合通道注意力特征，得到所述目标眼睛图像的融合空间注意力特征；

42、根据所述目标检测网络处理所述融合空间注意力特征，得到所述目标眼睛图像中所述目标瞳孔的瞳孔中心点坐标和瞳孔尺寸。

43、可选的，所述目标眼睛图像包括所述眼底相机拍摄得到的两帧目标眼睛图像，所述眼底相机为双目同步眼底相机；

44、所述控制单元根据所述瞳孔中心点坐标和所述瞳孔尺寸，控制眼底相机移动至所述瞳孔中心点坐标对应的目标位置时，具体用于：

45、根据两帧所述目标眼睛图像的所述瞳孔中心点坐标和所述瞳孔尺寸，确定所述瞳孔中心点坐标对应的目标位置；

46、根据所述目标位置确定所述眼底相机的移动方向和移动距离；

47、根据所述移动方向和所述移动距离控制步进电机，使所述眼底相机的中心点被所述步进电机驱动而移动至所述目标位置。

48、本方案的有益效果在于：

49、在基于瞳孔识别模型识别目标眼睛图像的瞳孔中心点坐标时，基于通道注意力模块和空间注意力模块来处理目标眼睛图像，从而结合目标眼睛图像在通道和空间两个维度的图像特征进行瞳孔中心点识别，使瞳孔识别模型能够更全面的关注目标眼睛图像在多个维度上的重要特征，有效提高了识别结果的准确性，进而提高了眼底相机根据瞳孔中心点拍摄眼底图像的成功率。