[发明专利]一种图像处理方法、装置、电子设备及计算机可读存储介质

权利要求书

1.一种图像处理方法，其特征在于，所述方法包括：

获取待处理图像；

将所述待处理图像进行对比度平衡处理，得到目标图像；

确定所述目标图像中的缺陷信息，其中，所述缺陷信息表征所述目标图像对应的设备的缺陷；

其中，所述确定所述目标图像中的缺陷信息的步骤，包括：

将所述目标图像输入至训练好的初步判图模型中；

所述初步判图模型输出所述目标图像的缺陷信息；

其中，所述初步判图模型输出所述目标图像的缺陷信息的步骤，包括：

所述初步判图模型对所述目标图像中的缺陷信息进行识别；

输出多个识别结果，每种所述识别结果对应一种缺陷信息，所述缺陷信息包括缺陷位置和缺陷类型；

确定各所述识别结果的置信度；

获取置信度最高的识别结果，作为所述目标图像的缺陷信息并输出；

其中，所述目标图像的缺陷信息包括第一缺陷信息、第二缺陷信息、第三缺陷信息，

所述初步判图模型输出第一缺陷信息对应的第一置信度，第二缺陷信息对应的第二置信度，第三缺陷信息对应的第三置信度；

获取第一置信度、第二置信度、第三置信度中最高的置信度对应的缺陷信息，作为目标图像中设备的缺陷信息输出；

其中，所述初步判图模型包括以下任意之一：

Faster-rcnn模型和yolo模型；

所述方法还包括：

获取所述初步判图模型的训练集；

基于所述训练集对所述初步判图模型进行训练；

所述获取所述初步判图模型的训练集的步骤，包括：

从机台获取携带有缺陷标识的图像数据；

采用对比度均衡算法对各所述图像数据进行处理，得到各目标图像数据；

对每个所述目标图像数据标注缺陷位置和缺陷类型；

将携带有所述缺陷位置和所述缺陷类型的各所述目标图像数据，作为所述初步判图模型的训练集；

所述方法还包括：

计算每个目标图像数据与对应的待调整目标图像数据的损失惩罚项信息，其中，待调整目标图像数据为未标注缺陷位置和缺陷类型的图像数据；将损失惩罚项信息输入至初步判图模型，以获得更新后的初步判图模型；返回执行计算每个目标图像数据与对应的待调整目标图像数据的损失惩罚项信息的步骤，直到最新的损失惩罚项信息收敛；

其中，所述计算每个所述目标图像数据与对应的待调整目标图像数据的损失惩罚项信息的步骤，包括：

计算每个所述目标图像数据的缺陷位置与待调整目标图像数据的预测缺陷位置的第一损失惩罚项信息；

计算该所述目标图像数据的缺陷类型与待调整目标图像数据的预测缺陷类型的第二损失惩罚项信息；

基于所述第一损失惩罚项信息和所述第二损失惩罚项信息，得到所述目标图像数据与对应的待调整目标图像数据的损失惩罚项信息。

2.根据权利要求1所述的方法，其特征在于，所述将所述待处理图像进行对比度平衡处理，得到目标图像的步骤，包括：

调用OPENCV库中的对比度平衡算法；

基于所述对比度平衡算法对所述待处理图像进行处理，得到目标图像。