中文

Semmeldetector:机器学习在商业面包店中的应用

计算机视觉与模式识别 2024-06-07 v1

摘要

Semmeldetector是一个机器学习应用,利用目标检测模型对图像中的烘焙食品进行检测、分类和计数。我们的应用使商业面包师能够追踪未售出的烘焙食品,从而优化生产并提高资源效率。我们编制了一个包含1151张图像的数据集,区分18种不同类型的烘焙食品,用于训练检测模型。为促进模型训练,我们使用Copy-Paste增强流水线来扩充数据集。我们在检测任务上训练了最先进的目标检测模型YOLOv8。我们测试了不同训练数据、模型规模和在线图像增强流水线对模型性能的影响。我们整体表现最佳的模型在测试集上达到了89.1%的[email protected]。基于我们的结果,我们得出结论:即使在面包店等意想不到的行业,机器学习也可以成为有价值的工具,即使数据集非常有限。

关键词

引用

@article{arxiv.2406.04050,
  title  = {Semmeldetector: Application of Machine Learning in Commercial Bakeries},
  author = {Thomas H. Schmitt and Maximilian Bundscherer and Tobias Bocklet},
  journal= {arXiv preprint arXiv:2406.04050},
  year   = {2024}
}