一种无锚点框的三维肺结节检测模型训练方法及装置
基本信息
| 申请号 | CN202110248609.0 | 申请日 | - |
| 公开（公告）号 | CN112614133B | 公开（公告）日 | 2021-07-06 |
| 申请公布号 | CN112614133B | 申请公布日 | 2021-07-06 |
| 分类号 | G06T7/00 | 分类 | 计算；推算；计数; |
| 发明人 | 杜强;王思其;郭雨晨;聂方兴;唐超 | 申请（专利权）人 | 北京小白世纪网络科技有限公司 |
| 代理机构 | 北京八月瓜知识产权代理有限公司 | 代理人 | 秦莹 |
| 地址 | 100083 北京市海淀区王庄路1号院清华同方科技大厦D座19层1903-1号 | ||
| 法律状态 | - | ||
摘要
| 摘要 | 本发明公开了一种无锚点框的三维肺结节检测模型训练方法及装置，该方法包括：获取数据集中三维肺结节样本的图像数据、标注类别和标注坐标点并构建三维肺结节数据库；将进行数据增强后的图像数据输入到卷积神经网络进行处理，得到多个特征图；将特征图输入到预先设置的具有两个卷积分支的共享头部网络，输出预测类别、预测中心度和预测坐标点，计算预测置信度和标注中心度，将标注坐标点、预测坐标点、预测置信度、标注类别、预测中心度、标注中心度输入到损失函数，反向传播，训练模型。采用本发明实施例，解决了有锚点框所带来的问题、减少了很多的计算量，采用双分支的网络结构使得不同任务间的特征不互相干扰，模型结果更加准确。 |
