摘要
当前,绝大多数工地清点钢筋等材料使用传统人工方式,人工清点方式耗费人力且容易造成数量统计错误。为解决上述问题,设计了一个基于AI技术的钢筋数量智能识别模型。根据实际环境中拍摄的钢筋图片作为样本集,使用LabelImg标注工具进行数据标注,利用YOLOv3算法训练识别模型,使用PyTorch作为机器学习平台。本系统使用YOLOv3算法构造神经网络进行深度学习从而训练模型,使用Loss函数、mAP函数、计算F1值等多个参数进行评估模型,利用训练好的模型绘制框图预测钢筋位置以及计算出图中的钢筋总数。经测试,识别率为97.83%。
- 单位