我们做了一个智能零售结算平台

Datawhale

共 3647字,需浏览 8分钟

 · 2022-07-04

 Datawhale分享 
作者:颜鑫,Datawhale成员

目前在零售行业的实际运营过程中,会产生巨大的人力成本,例如导购、保洁、结算等,而其中,尤其需要花费大量的人力成本和时间成本在识别商品并对其进行价格结算的过程中,并且在此过程中,顾客也因此而需要排队等待。这样一来零售行业人力成本较大、工作效率极低,二来也使得顾客的购物体验下降。

随着计算机视觉技术的发展,以及无人化、自动化超市运营理念的提出,利用图像识别技术及目标检测技术实现产品的自动识别及自动化结算的需求呼之欲出,即自动结账系统(Automatic checkout, ACO)。基于计算机视觉的自动结账系统能有效降低零售行业的运营成本,提高顾客结账效率,从而进一步提升用户在购物过程中的体验感与幸福感。

AI结算的核心是图像识别。图像识别的准确率决定了AI结算落地的可行性。目前,AI结算面临着如下几个痛点问题:

1.商品包装相似:同类别商品口味不同且价格也不同,不同类别商品外包装相似,都对图像识别精度具有较高的要求;
2.干扰因素众多:同类别商品在识别时容易因角度问题发生变形、折叠、遮挡等干扰,对识别结果造成影响;
3.品类更新极快:零售商品通常以小时级别速度更新迭代,每增加新产品时若仅靠单一模型均需重新训练模型,模型训练成本及时间成本极大;
4.系统性能要求高:需要同时解决检测和识别两个任务,选模型和优化时要权衡精度与速度两方面。

袋鼯麻麻ColugoMum致力于为中小型型线下零售体验店提供基于视觉的智能零售结算方案,并在Github社区和启智社区同步开源。

https://github.com/ColugoMum
https://git.openi.org.cn/ColugoMum

基于上述痛点问题,ColugoMum团队采用飞桨PaddleClas[1]团队提出的PP-ShiTu[2]图像识别系统。基于PP-ShiTu实现的商品识别方案为零售场景中商品多类别、小样本、高相似和更新频繁问题提供了新的思路,不仅能对多类别商品进行精准识别,也可以满足对预测效率的极致追求。尤为实用的功能在于:实际上线使用的时候,遇到新的需要识别的商品类别,无需重新训练模型,只需要在检索库中增加该类别图像特征,就能够实现对新商品的识别!

PP-ShiTu是一个实用的轻量级通用图像识别系统,主要由主体检测、特征学习和向量检索三个模块组成。该系统从骨干网络选择和调整、损失函数的选择、数据增强、学习率变换策略、正则化参数选择、预训练模型使用以及模型裁剪量化8个方面,采用多种策略,对各个模块的模型进行优化,并经过10w+类别数据进行训练,最终得到在CPU上预测时间仅需0.2s的多场景通用图像识别系统。

简单来说,PP-ShiTu的使用分为三步:

1.通过主体检测模型,对图片中的物体一一识别;
2.对每个候选区域进行特征提取;
3.将特征提取后的向量在检索库中进行检索,完成匹配,返回识别结果。

考虑到实际零售场景对于精度和预测速度的极致追求,ColugoMum团队在主体检测部分选取了PicoDet模型作为主体检测算法,选取了轻量级PPLCNet_x2_5_ssld模型用作特征提取,最后使用向量搜索模块Faiss中的HNSW32作为检索算法,实现速度与精度的极致平衡。

基于此,ColugoMum团队基于RP2K数据集已经实现了最高**96.91%**的预测精度。

RP2K数据集[3]收录了50万+张零售商品货架图片,商品种类超过2,000种,是目前零售类数据集中产品种类数量最多的数据集。不同于一般聚焦新产品的数据集,RP2K收录了超过50万张零售商品货架图片,商品种类超过2000种,该数据集是目前零售类数据集中产品种类数量TOP1,同时所有图片均来自于真实场景下的人工采集,针对每种商品,品览提供了十分详细的标注。

此外,ColugoMum也收集整理了业内SKU级别的商品图像数据集,并期待和开发者们一道, 开源出能够在业内有影响力、符合实际场景应用需求的数据集。

https://github.com/ColugoMum/Dataset

并且,ColugoMum团队开源了基于RP2K数据集的高精度训练模型和预测模型。开发者可以在提供的训练模型上基于自己的数据进行微调,也可以使用提供的预测模型直接进行预测体验。同时,ColugoMum也开启了基于RP2K的打榜活动,欢迎开发者们参与。

https://github.com/ColugoMum/Exprements

部署方面使用飞桨服务化部署框架Paddle Serving[4]进行部署,满足用户批量预测、数据安全性高、延迟低的需求,在CPU上仅需0.2秒即可实现预测效果,真正做到预测速度与精度的极致平衡。

为了方便开发者们更好地理解PP-ShiTu、更好地利用其在图像识别领域的优势,ColugoMum团队开源了基于图像识别的智慧零售商品识别教程,开发者可以在此基础上使用PP-ShiTu快速对接业务。

https://github.com/ColugoMum/Goods_Recognition

基于上述核心技术,目前ColugoMum团队利用PP-ShiTu技术,开源了云边一体、符合实际场景应用需求的商品识别Smart_container。其能够精准地定位顾客购买的商品,并进行智能化、自动化的价格结算。

https://github.com/ColugoMum/Smart_container

当顾客将自己选购的商品放置在制定区域内时,Smart_container能够精准地定位识别每一个商品,并且能够返回完整的购物清单及顾客应付的实际商品总价格。而当系统有新商品增加时,只需更新检索库即可,无需重新训练模型。Smart_container覆盖硬件结算台、小程序管理平台、大数据可视化平台,实现了多端统一,智慧管理。

核心开发团队

颜鑫,华东理工大学自动化专业大三在读,研究方向为多机器人的协同控制与决策,主要兴趣点为计算机视觉、强化学习、推理部署。飞桨开发者技术专家、Datawhale成员、华为云享专家、国家级大学生创新创业项目第一主持人,曾获第十三届“挑战杯”大学生创业计划竞赛上海市铜奖、华东理工大学第十届“奋进杯”大学生创业计划竞赛金奖、拥有两篇软件著作权、一篇国际会议论文。

沈晨,华东理工大学信息工程专业大三在读,曾获得 CRAIC 中国机器人及人工智能大赛上海市二等奖、上海市大学生计算机应用能力设计大赛上海市二等奖,华东理工大学第十届“奋进杯”大学生创业计划竞赛金奖,参与多个大型优秀开源项目,拥有软件著作权两篇,曾在IEEE国际会议发表论文一篇,另有一项实用新型专利正在受理。曾任信息学院团委组织部副部长,获得校优秀奖学金和优秀学生先进称号。

杜旭东,华东理工大学信息工程专业大三在读,熟练掌握C/C++/Python/Jave/Matlab/verilog等多种编程语言,上海市大学生计算机应用能力设计大赛上海市二等奖,参与多个大型优秀开源项目,拥有软件著作权两篇,一项实用新型专利正在受理。曾任信息学院社团管理部副部长,获得校优秀奖学金。

在这里,我们向曾参与ColugoMum项目研发的同学表示感谢:黄小悦、王鑫、赵祎安、周天奕、申佳川等,向为ColugoMum提供帮助的Datawhale组织飞桨社区启智社区表示感谢! 我们欢迎更多开发者共同参与零售商品识别数据集、商品识别打榜以及Smart Container的开源共建活动,共同促进AI开源开放生态体系建设,推动我国实体零售向数字化、智能化方向转型发展。

未来,ColugoMum团队会不断打破产品和技术边界,依托开源社区,开源出更多更好的、能真正赋能实体零售的开源项目,真正推动我国实体零售向智能化、数字化转型,实现ColugoMum“降本增效、赋能零售”的使命。

参考文献 :

1.https://github.com/PaddlePaddle/PaddleClas  

2.S. Wei et al., "PP-ShiTu: A Practical Lightweight Image Recognition System," arXiv preprint arXiv:2111.00775, 2021. 

3.J. Peng, C. Xiao, and Y. Li, "RP2K: A large-scale retail product dataset for fine-grained image classification," arXiv preprint arXiv:2006.12634, 2020. 

4.https://github.com/PaddlePaddle/Serving

开源分享,点赞三连
浏览 83
点赞
评论
收藏
分享

手机扫一扫分享

举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

举报