图像分类是整个计算机视觉领域中最基础的任务,也是最重要的任务,所以这次我通过一个对人脸进行表情识别的任务,来向大家展示一个完整的图像分类任务的流程。
内容包含数据集制作,模型训练,模型测试三个完整部分,需要打包代码的可以留下邮箱,文章中的代码也是全的并分类做了介绍,代码也都是测试过的,可以正确运行,代码基本都加了注释。
文章目录
一、爬取数据集
由于没有对应的开源数据集,而且对于微笑这类的表情图片更是难于找到对应的数据集,但是搜索引擎上却有海量的图片,所以我们可以从上面爬取出我们需要的图片。我们可以自己编写爬虫,也可以找一些现成的大佬编写的爬虫代码直接使用。
推荐:https://github.com/sczhengyabin/Image-Downloader
只需输入关键词,选定浏览器就可以爬取所需要的图片,这里输入的关键词是 “微笑”
二、数据集整理
可以看到爬取的数据集是非常脏的,体现在后缀格式,非人脸图片,名称