好久没有写东西了，日常在实验室搞别的东西渐渐怠慢了更新，实在有些惭愧。每天过得如咸鱼一般，虽然还有作业、有任务，但生活的享受还是不能丢失。每天最忙最累的时候，必须要有一杯热腾腾的星巴克，哪怕日子穷困潦倒，对生活的热爱也不能减少。最近有一个有趣的图像集，叫10-monkey-species，是十个种类的猴子的图像集。从kaggle上搜10 Monkey Species便可以把数据集给下载下来，解压，显示如下：
训练集和验证集文件夹分别都有10类猴子的图像，每一类都有一百多张图像
使用tensorflow库中的keras，对数据集进行归一化、图像增强等预处理
再用keras来编写几层神经网络，代码如下：
模型训练过程的损失值以及准确率的输出，还要变化曲线的显示
可以看出，在300步以后，模型在训练集的准确率有97%，在测试集的准确率有84%左右。模型的准确率还是不错的，但是还可以继续提升。使用迁移学习的方式，调用已经训练好的resnet50这一网络的模型，使用本次的图像集输入模型中进行微调，便可以在本次验证图像集中达到更高的准确率了。下篇文章再搞一下这个吧，这次有些累了，继续咸鱼一下吧。
在本人公众号回复「猴子」可获取完整代码。
网友评论