本文为Udacity优达学城深度学习课程笔记第三篇,使用图像增强对猫狗图像进行分类。
课程地址:https://classroom.udacity.com/courses/ud187
最终目的是训练CNN模型,使其能识别上面的小狗狗是小狗狗。可能遇到的问题有过拟合、输入图像的尺寸不一,不过不用担心,本文会一一解决。为了保持笔记与课程内容一致,代码没有进行修改,如需提升识别准确率可自行对模型进行修改。
import os
import matplotlib.pyplot as plt
import numpy as np
import t