结合SSD及Resnet50网络的电力箱图片锈蚀破损识别方法

摘要

本发明公开了一种结合SSD及Resnet50网络的电力箱图片锈蚀破损识别方法。采集电力箱样本图片，预处理并进行筛选整理并标注，形成训练集；将训练集输入到SSD目标检测网络中预测输出电力箱小图，训练得到电力箱检测模型；利用电力箱小图输入到Resnet50分类网络中预测判断出电力箱是否存在锈蚀破损情况，训练得到锈蚀破损检测模型；通过测试后结合形成电力箱锈蚀破损识别综合模型，利用电力箱锈蚀破损识别综合模型实现对待测电力箱图片进行检测。本发明能够较好地检测出电力箱在图像中的位置并判断是否出现锈蚀破损等状况和结果，解决了电力箱拍摄图像背景干扰大、锈蚀破损情况判断困难等问题。