项目由图像采集、模型选择、模型训练、模型推理、结果显示5个部分组成。使用光学相机采集隧道螺栓图像,并对采集的图像进行旋转校正、平移放缩、对比度调整等规范为640*640大小的图像,构成数据集原始图像。采集的数据集以VOC的格式构建数据集,按照一定比例划分训练集、验证集。模型可选择不同主干网络构成的UNet模型,并加载相应预训练权重。将数据集输入到模型中进行训练,训练结束后将保留最优的权重文件。将待检测图片放入指定文件夹下,模型进行推理时将加载最优模型进行推理,模型的预测结果可以保存为背景为黑色,目标区域为白色的二值图像,也可以与原图进行融合,将预测结果在原图上显示。声明:本文仅代表作者观点,不代表本站立场。如果侵犯到您的合法权益,请联系我们删除侵权资源!如果遇到资源链接失效,请您通过评论或工单的方式通知管理员。未经允许,不得转载,本站所有资源文章禁止商业使用运营!
![](https://stacdn.proginn.com/image/common/qrcode/download.png?v=1)
下载安装【程序员客栈】APP
实时对接需求、及时收发消息、丰富的开放项目需求、随时随地查看项目状态
评论