Image classification with FastAI

之前一篇文章介绍了FastAI v0.7中处理dogscats数据集的过程。
这一篇则介绍一下FastAI v1.0版本变化后的dogscats图像分类的使用方法。相比较v0.7，v1.0整体来说api更加简洁清晰一些，特别是预测和评估的部分。

让我们开始吧

1.配置colab中的FastAI环境

!curl https://course.fast.ai/setup/colab | bash

2.导入包

from fastai import *
from fastai.vision import *

3.下载数据集文件

!wget http://files.fast.ai/data/dogscats.zip

4.解压数据集

!unzip dogscats.zip

5.加载数据集

path = "/content/dogscats/"
data = ImageDataBunch.from_folder(path,ds_tfms=get_transforms(), size=224,bs=32).normalize(imagenet_stats)

6.检查数据文件

data.show_batch(3,figsize=(8,8))

7.设置深度学习中的神经网络，选择预置的resnet50

learner = create_cnn(data,models.resnet50,metrics=[accuracy])

8.显示训练结果

learner.fit(1)

这里需要稍等片刻，花了差不多8分半钟。

9.显示网络结构

print(learner.summary())

10.预测单张图片的结果

img = data.train_ds[0][0]
img.show()
learner.predict(img)

这里也可以使用upload方法上传本地图片来测试
v1.0中预测单张图片的方法似乎更加简单方便了。可以直接把图片传入predict方法中。

11.制作混淆矩阵

interp = ClassificationInterpretation.from_learner(learner)
interp.plot_confusion_matrix()
plt.grid(False)

发现可以通过plt.grid(False)用来取消图中的网格线，发现不少文章中的贴图都是带有网格线，十分影响美观。^^

参考：
https://github.com/adi0229/ML-DL/blob/master/dogcat_fastaiv3.ipynb