公开(公告)号：CN118628748A

公开(公告)日：2024-09-10

申请号：CN202410888503.0

申请日：2024-07-04

申请人： 福建省高速公路科技创新研究院有限公司 ,  福州大学

发明人： 曾俊铖 ,  陈智威 ,  吴少峰 ,  王峥 ,  谷音 ,  周森辉 ,  林文昊 ,  林龙威 ,  杨志辰

IPC分类号： G06V10/26 ,  G06V10/25 ,  G06V10/44 ,  G06V10/52 ,  G06V10/80 ,  G06V10/82 ,  G06V10/774 ,  G06V10/776 ,  G06N3/0464 ,  G06N3/045 ,  G06N3/048

摘要： 本发明公开一种基于改进YOLOv5网络模型的钢筋图像分割方法、产品、介质及设备，涉及图像识别领域。本发明首先获取钢筋图像数据集；利用C3R模块和CBAM模块改进YOLOv5算法网络结构，建立改进YOLOv5网络模型；利用钢筋图像数据集训练改进YOLOv5网络模型，训练完成后得到钢筋图像分割模型；将待分割的钢筋图像进行透视变换，得到变换后钢筋图像；将变换后钢筋图像输入钢筋图像分割模型，分割出图像中的钢筋。采用本发明方法能够提高钢筋图像分割提取的准确度。