是否有可用于对 30,000 种不同工具和零件进行分类的 AI 服务?

数据挖掘 神经网络 分类 图像分类 图像识别
2022-03-05 07:58:35

我正在尝试构建一个图像分类器,人们可以在其中拍摄工具或零件的照片并对图像进行分类。很像 bixby 或亚马逊的工具做类似的事情,但只有 30,000 种不同的物品需要分类,而不是庞大的购物目录。

我希望有一个 AI 服务可以完成大量工作,因为我试图避免训练我自己的 CNN,因为我的项目无法实现拍摄我需要的大量照片所需的人力。我绝对可以得到人力来拍一些照片,大概 300,000 到 100 万张,但我认为在训练一个全新的 CNN 时这还不够。

1个回答

使用预训练模型和迁移学习。

keras为例:

https://keras.io/applications/

你只需要为每个类别准备几十张图片,就可以有相当不错的表现。