一种基于机器视觉的优化废钢图像的AI生成方法

本发明涉及冶金领域和图像识别领域，尤其涉及一种基于机器视觉的优化废钢图像的AI生成方法，包括获取废钢卸料场景下带有废钢料的图像帧数据；对带有废钢料的图像帧数据进行预处理，添加噪声干扰项，按比例随机划分成数据集；建立生成对抗模型，并对对抗模型进行训练，获取训练后的生成模型；将实时获取的卸料场景下视频流图像帧数据输入至训练后的生成模型；将训练后的生成模型输出的结果发送给废钢图像优化系统。本发明的优点是：网络模型基于深度学习生成对抗式网络技术，在废钢卸料现场卸车场景中，实现了自动生成不同环境的废钢料图像以及去噪、去模糊的抓拍图像，改进的生成对抗式网络模型针对极端情况样本集短缺的需求。