胃部病灶识别模型训练方法、胃部病灶识别方法

摘要

本发明公开一种胃部病灶识别模型训练方法、胃部病灶识别方法及存储介质，应用于胶囊内窥镜，该模型训练包括：标注样本图像的病灶区域的轮廓及病灶标签，得到训练样本集；将该训练样本集输入初始模型，该初始模型包括输入端、主干网络、颈部网络和预测网络；通过该输入端对该训练样本集进行预处理，得到处理后的训练样本集；通过该主干网络提取该处理后的训练样本集的特征；通过该颈部网络增强该处理后的训练样本集的该特征；通过该预测网络进行胃部病灶的分类和预测框的回归，得到胃部病灶识别模型。本发明实施例采用Yolov5模型作为初始模型，经训练后得到的胃部病灶识别模型预测精度高，可以实现对胃部病灶的自动、精准识别。