摘要
本申请涉及一种变电站多目标设备检测模型生成方法、装置和计算机设备。所述方法包括:将设备图像输入至待训练的目标检测模型中的输入层,得到特征图,并按照预设的划分方式,将特征图划分为第一特征图集合和第二特征图集合;将第一特征图集合输入下采样模块进行下采样处理,得到下采样结果,将下采样结果集合输入上采样模块进行最近邻插值处理,得到上采样结果;根据第二特征图集合和上采样结果,确定设备图像检测结果;根据设备图像检测结果和针对设备图像预先标注的设备标注信息,训练目标检测模型,并对目标检测模型进行剪枝,得到变电站多目标设备检测模型。采用本方法能够提高变电站多目标设备检测的检测效率。
技术关键词
变电站
检测模型生成方法
图像
采样模块
上采样
计算机设备
模型生成装置
计算机程序产品
处理器
纹理特征
通道
输入模块
识别模块
可读存储介质
存储器
像素
参数
批量