项目由图像采集、模型选择、模型训练、模型推理、结果显示5个部分组成。使用光学相机采集隧道螺栓图像，并对采集的图像进行旋转校正、平移放缩、对比度调整等规范为640*640大小的图像，构成数据集原始图像。采集的数据集以VOC的格式构建数据集，按照一定比例划分训练集、验证集。模型可选择不同主干网络构成的UNet模型，并加载相应预训练权重。将数据集输入到模型中进行训练，训练结束后将保留最优的权重文件。将待检测图片放入指定文件夹下，模型进行推理时将加载最优模型进行推理，模型的预测结果可以保存为背景为黑色，目标区域为白色的二值图像，也可以与原图进行融合，将预测结果在原图上显示。