发明公开
- 专利标题: 一种基于轻量化深度神经网络的建造物料视觉盘存方法
-
申请号: CN202310836296.X申请日: 2023-07-07
-
公开(公告)号: CN117197649A公开(公告)日: 2023-12-08
- 发明人: 邓曦 , 刘寿松 , 王楷 , 钱琪 , 杨应铭 , 曹馨雨 , 刘健
- 申请人: 中国建筑科学研究院有限公司 , 中国建筑技术集团有限公司 , 重庆大学
- 申请人地址: 北京市朝阳区北三环东路30号
- 专利权人: 中国建筑科学研究院有限公司,中国建筑技术集团有限公司,重庆大学
- 当前专利权人: 中国建筑科学研究院有限公司,中国建筑技术集团有限公司,重庆大学
- 当前专利权人地址: 北京市朝阳区北三环东路30号
- 代理机构: 重庆强大凯创专利代理事务所
- 代理商 向林
- 主分类号: G06V20/10
- IPC分类号: G06V20/10 ; G06V10/44 ; G06V10/80 ; G06V10/764 ; G06V10/82 ; G06N3/0475 ; G06N3/045 ; G06N3/094 ; G06N3/0464 ; G06N3/048 ; G06N3/08
摘要:
本发明涉及计算机视觉技术领域,具体涉及一种基于轻量化深度神经网络的建造物料视觉盘存方法,包括:采集建筑物物料数据集,通过人工结合半自动标注对钢筋数据集进行标注,并将钢筋数据集划分为训练集与测试集;基于Improved ShufflNet v2设计轻量化神经网络模型;利用构建的训练集对轻量化神经网络模型进行训练,并用测试集对轻量化神经网络模型性能进行初步测试;对训练后的轻量化神经网络模型基于BN层进行通道剪枝,简化轻量化神经网络模型,并再次用测试集测试剪枝后的轻量化神经网络模型性能;利用训练后并剪枝的轻量化神经网络模型对钢筋图像进行检测,得到钢筋数量。本发明使得模型精简,降低对处理器计算能力要求,使其可在移动端部署。