息肉检测模型的训练方法、息肉检测方法及相关装置
摘要:
本公开涉及一种息肉检测模型的训练方法、息肉检测方法及相关装置,以提高息肉检测模型在检测低对比度的内窥镜图像的息肉检测性能。该训练方法包括:将目标样本图像输入息肉检测模型,得到图像特征、多个预测框特征以及息肉预测结果;根据息肉预测结果确定图像特征对应的特征权重,将特征权重与图像特征相乘得到目标图像特征,根据目标图像特征、正样本图像和负样本图像确定第一损失值;根据息肉预测结果、预测框特征、正样本框特征和负样本框特征,确定第二损失值;根据息肉预测结果与目标样本图像标注的息肉标签确定第三损失值;根据第一损失值、第二损失值和第三损失值,调整息肉检测模型的参数。
0/0