公开(公告)号：CN115829939A

公开(公告)日：2023-03-21

申请号：CN202211393972.2

申请日：2022-11-08

Applicant: 中国科学院自动化研究所

Inventor： 周小虎 ,  陶可欣 ,  谢晓亮 ,  刘市祺 ,  李芮麒 ,  奉振球 ,  侯增广 ,  于喆 ,  李思远 ,  张鑫

IPC: G06T7/00 ,  G06V10/46 ,  G06V10/25 ,  G06V10/774 ,  G06N3/08 ,  G06N3/048 ,  G06N3/0464

Abstract: 本发明提供一种多导丝端点定位模型建立方法和多导丝端点定位模型。方法包括：建立初始多导丝端点定位模型；建立导丝定位数据集，导丝定位数据集中包括图像及图像对应的标签，标签包括检测框信息和导丝端点信息；采用训练集对初始多导丝端点定位模型进行训练，训练集包括从数据集抽取的的图像及对应的标签，训练时以图像作为输入，输出得到检测框估计信息和导丝端点估计信息；根据训练集中图像对应的检测框信息与检测框估计信息之间的误差以及导丝端点信息与导丝端点估计信息之间的误差建立损失函数；优化损失函数，更新初始多导丝端点定位模型的参数，直至收敛，以得到多导丝端点定位模型。该方法可以端对端对模型进行训练且特征提取不冗余。