PyTorch:基于FastAI的猫狗识别

背景

上次自己用PyTorch对AlexNet进行猫狗识别取得了不错效果。后续我又继续使用Resnet50进行猫狗分类,但是却没有取得较好的效果,想来还是自己炼丹的技巧太差了。因此,后续决定使用FastAI进行训练(FastAI封装了很多炼丹技巧),最后效果还不错,Kaggle的分数为0.05779,已经可以挤进前100了;相比上次,效果还是很不错的。
在这里插入图片描述


代码(基本抄的参考链接)

https://github.com/Yannnnnnnnnnnn/learnPyTorch/blob/master/catanddog_fastai.ipynb


参考链接

https://www.kaggle.com/toshikazuwatanabe/fast-ai-latest-dogs-vs-cats

你可能感兴趣的:(PyTorch)