基于深度学习和迁移学习的水果图像分类

作者：廉小亲; 成开元; 安飒; 吴叶兰; 关文洋

来源：测控技术, 2019, 38(06): 15-18.

DOI：10.19708/j.ckjs.2019.06.004

• 图像识别
• 深度学习
• Softmax
• 迁移学习

图像识别作为深度学习领域内的一项重要应用,水果图像的分类识别在智慧农业以及采摘机器人等方面具有重要应用。针对以往传统图像分类算法存在泛化能力差、准确率不高等问题,提出一种在TensorFlow框架下基于深度学习和迁移学习的水果图像分类算法。该算法采用Inception-V3的部分模型结构对水果图像数据进行特征提取,采用Softmax分类器对图像特征进行分类,并通过迁移学习方式进行训练得到迁移训练模型。测试结果表明,该算法与传统水果分类算法对比,具有较高识别准确率。

• 单位
  北京工商大学