Jun, 2024

Semmeldetector: 商业烘焙业中的机器学习应用

TL;DR使用机器学习应用程序 Semmeldetector,可以利用目标检测模型在图像中检测、分类和计数烘焙食品。我们的应用程序允许商业烘焙师跟踪未售出的烘焙食品,从而优化生产并提高资源效率。通过编制包含 1151 张图像的数据集,可以区分出 18 种不同类型的烘焙食品来训练我们的检测模型。为了促进模型训练,我们使用了一个复制粘贴的增强流水线来扩展我们的数据集。我们在检测任务中使用了最先进的目标检测模型 YOLOv8 进行训练。我们测试了不同训练数据、模型规模和在线图像增强流水线对模型性能的影响。我们最佳的模型在测试集上达到了 89.1% 的 [email protected]。基于我们的结果,我们得出结论,机器学习即使在像烘焙行业这样的未预见行业中,也可以成为有价值的工具,即使数据集非常有限。