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

测控技术

模式识别与人工智能２０１９３８６·１５·

年第卷第期

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

廉小亲，成开元，安飒，吴叶兰，关文洋

（北京工商大学计算机与信息工程学院食品安全大数据技术北京市重点实验室，北京１０００４８）

摘要图像识别作为深度学习领域内的一项重要应用水果图像的分类识别在智慧农业以及采摘机器人

：，

等方面具有重要应用。针对以往传统图像分类算法存在泛化能力差、准确率不高等问题，提出一种在

ＴｅｎｓｏｒＦｌｏｗＩｎｃｅｐｔｉｏｎＶ３

框架下基于深度学习和迁移学习的水果图像分类算法。该算法采用的部分模

型结构对水果图像数据进行特征提取采用分类器对图像特征进行分类并通过迁移学习方式

，Ｓｏｆｔｍａｘ，

进行训练得到迁移训练模型测试结果表明该算法与传统水果分类算法对比具有较高识别准确率

。，，。

：；；；

关键词图像识别深度学习Ｓｏｆｔｍａｘ迁移学习

中图分类号：ＴＰ２９文献标识码：Ａ文章编号：１０００－８８２９（２０１９）０６－００１５－０４

ｄｏｉ：１０．１９７０８／ｊ．ｃｋｊｓ．２０１９．０６．００４

ＦｒｕｉｔＩｍａｇｅＣｌａｓｓｉｆｉｃａｔｉｏｎＢａｓｅｄｏｎＤｅｅｐＬｅａｒｎｉｎｇａｎｄＴｒａｎｓｆｅｒＬｅａｒｎｉｎｇ

ＬＩＡＮＸｉａｏｑｉｎ牞ＣＨＥＮＧＫａｉｙｕａｎ牞ＡＮＳａ牞ＷＵＹｅｌａｎ牞ＧＵＡＮＷｅｎｙａｎｇ

牗ＢｅｉｊｉｎｇＫｅｙＬａｂｏｒａｔｏｒｙｏｆＢｉｇＤａｔａＴｅｃｈｎｏｌｏｇｙｆｏｒＦｏｏｄＳａｆｅｔｙ牞ＳｃｈｏｏｌｏｆＣｏｍｐｕｔｅｒａｎｄＩｎｆｏｒｍａｔｉｏｎＥｎｇｉｎｅｅｒｉｎｇ牞

ＢｅｉｊｉｎｇＴｅｃｈｎｏｌｏｇｙａｎｄＢｕｓｉｎｅｓｓＵｎｉｖｅｒｓｉｔｙ牞Ｂｅｉｊｉｎｇ１０００４８牞Ｃｈｉｎａ牘

Ａｂｓｔｒａｃｔ牶Ｉｍａｇｅｒｅｃｏｇｎｉｔｉｏｎｉｓａｎｉｍｐｏｒｔａｎｔａｐｐｌｉｃａｔｉｏｎｉｎｔｈｅｆｉｅｌｄｏｆｄｅｅｐｌｅａｒｎｉｎｇ．Ｔｈｅｃｌａｓｓｉｆｉｃａｔｉｏｎａｎｄｒｅｃ

ｏｇｎｉｔｉｏｎｏｆｆｒｕｉｔｉｍａｇｅｓｈａｓｉｍｐｏｒｔａｎｔａｐｐｌｉｃａｔｉｏｎｓｉｎｓｍａｒｔａｇｒｉｃｕｌｔｕｒｅａｎｄｐｉｃｋｉｎｇｒｏｂｏｔｓ．Ｉｎｏｒｄｅｒｔｏｓｏｌｖｅｔｈｅ

ｐｒｏｂｌｅｍｓｏｆｐｏｏｒｇｅｎｅｒａｌｉｚａｔｉｏｎａｂｉｌｉｔｙａｎｄｌｏｗａｃｃｕｒａｃｙｉｎｔｒａｄｉｔｉｏｎａｌｉｍａｇｅｃｌａｓｓｉｆｉｃａｔｉｏｎａｌｇｏｒｉｔｈｍｓ牞ａｆｒｕｉｔ

ｉｍａｇｅｃｌａｓｓｉｆｉｃａｔｉｏｎａｌｇｏｒｉｔｈｍｂａｓｅｄｏｎｄｅｅｐｌｅａｒｎｉｎｇａｎｄｍｉｇｒａｔｉｏｎｌｅａｒｎｉｎｇｕｎｄｅｒＴｅｎｓｏｒＦｌｏｗｆｒａｍｅｗｏｒｋｉｓ

ｐｒｏｐｏｓｅｄ．ＴｈｅｐａｒｔｉａｌｍｏｄｅｌｓｔｒｕｃｔｕｒｅｏｆＩｎｃｅｐｔｉｏｎＶ３ｉｓｕｓｅｄｔｏｅｘｔｒａｃｔｔｈｅｆｅａｔｕｒｅｉｍａｇｅｏｆｆｒｕｉｔｉｍａｇｅ牞ａｎｄ

Ｓｏｆｔｍａｘｃｌａｓｓｉｆｉｅｒｉｓｕｓｅｄｔｏｃｌａｓｓｉｆｙｔｈｅｉｍａｇｅｆｅａｔｕｒｅｓ牞ｗｈｉｃｈｔｒａ