毛刺检测模型训练方法、装置、设备及存储介质

本申请提供一种毛刺检测模型训练方法、装置、设备及存储介质，该方法包括：采集电池极片图像数据，建立数据集；基于STDC网络构建编码器、解码器、损失函数计算模块和至少2个第一FCNHead模块的语义分割模型；采用数据集对语义分割模型进行训练得到毛刺检测模型；训练过程中，编码器的至少2个阶段处理模块的特征信息输入至对应的第一FCNHead模块，通过第一FCNHead模块输出第一预测结果信息至损失函数计算模块；解码器将输出的第二预测结果信息输出至损失函数计算模块；损失函数计算模块基于第一预测结果信息、第二预测结果信息判断语义分割模型是否训练完成。采用上述方案，可以提高毛刺检测模型的准确性。