图像分类是深度学习的典型问题,本文使用Mobile-net作为分类网络,实现自定义数据集的分类。主要特点是:
- 网上教程大多使用构造好的数据集如CIFAR10,缺少如何从头构建自己数据集的过程。本文从采集好的图片开始,一步步构造符合训练要求的数据集。
- 不熟悉深度学习与计算机视觉的工作者可能对种类繁多的网络结构感到困扰,编写起来更是麻烦。但是Pytorch的torchvision模块已经自带了很多SOTA网络的实现,因此本文直接import了Mobile-net,属于即插即用。
- 代码讲解较少,如果不熟悉Pytorch请先学习官方图像分类教程,了解基本概念。
综上,读者可以根据本教程快速实现自己的深度学习分类器。