玩家以AI帮忙分类乐高积木

对许多乐高玩家来说,拼各种模型很好玩,但是拆解后将各种积木分类却很累人。现在有人利用人工智能(AI)、Raspberry设计出可准确分类乐高积木的机器,以有效解决这个问题。

身兼软件工程师、游戏玩家及AI爱好者的Daniel West最近公布一段视频,显示以乐高积木和服务马达、相机等组件组合成名为“全能乐高分类机”(Universal Lego Sorting Machine)的机器。

West指出,他的乐高分类机是在看过2011年日本一个YouTuber的乐高分类项目及2017年第一个AI乐高分类机后产生的灵感。

“全能乐高分类机”利用10,000多个乐高积木、6个乐高马达及9个服务马达,并加上Raspberry Pi、相机、和纸做的输送带构成名为“Capture Unit”的机组。实际运行中,Raspberry Pi相机将输送带送来的乐高积木录像、截取图片后以无线发送到连接机器的Raspberry Pi机器处理,再送到邻近跑AI算法的服务器。后者利用算法分析识别积木形状,再将消息送回这台机器执行分类。

West利用真实的乐高积木、LDraw和Rebrickable 3D图片数据库,以及Google Tensorflow来训练其神经网络。他指出,Raspberry Pi的先天限制,使其无法跑太复杂的神经网络如YOLO或R-CNN。最后West运用计算机视觉和开源OpenCV library完成AI算法的开发。

这台机器可以将混杂的乐高积木堆,以约2秒1个的速度分到18个篮子。他说,它之所以称为“全能”,是因为它能识别3000种乐高组件,即使没见过的形状也能分辨。