训练可以识别物体材质可以用什么模型啊?

用一个材料数据集: MINC-2500
里面是各种材料物品的图片

img

img

用什么模型和方法训练出一个模型 可以实现: 识别图片中物品的材质?
比如:输入一张一个玻璃吊灯的图片,识别出来材质是:玻璃

如果你的数据集是标注好了的,这个可以使用图片分类算法解决:

  • 传统机器学习算法:特征描述算法(如sift,surf等)+SVM分类器
  • 深度学习算法(神经网络):VGG16,Resnet系列(推荐),Densent系列
    如果你想使用神经网络模型,可以看一下华为云的modelArts,上面有挺多图像分类、图像识别的模型的,很多都免费,可以看看能不能用得到