[发明专利]一种基于图像识别的零件分类方法

摘要

本发明公开了一种基于图像识别的零件分类方法，其属于图像处理领域，包括以下步骤：采集零件图像，将采集的零件图像进行预处理，并将预处理过后的零件图像作为训练集；构建零件识别神经网络，并对其网络参数进行初始化，得到初级零件识别神经网络；构建损失函数，以损失函数最小为目标，并通过训练集对初级零件识别神经网络进行训练，直至损失函数小于a，得到训练完成的零件识别神经网络；采集待识别图像，对待识别图像进行预处理，并将预处理后的待识别图像输入训练完成的零件识别神经网络中，得到零件分类结果。本发明通过实现零件分类，辅助维修人员或现场工作人员进行设备维修，避免了维修过程中零件识别费时费力的问题。