图像分类:搭建AlexNet并定义加载训练特定数据集
前言AlexNet模型是2012由Alex等人在ImageNet竞赛中提出,很优秀的机器学习分类的算法。也由此深度学习开始迅速发展,更多更深的神经网络被提出。在对一个简单图像分类任务下手时,我们要考虑到数据集的定义加载,然后搭建模型并对数据集进行训练测试。本文将讲一下其过程和遇到的一些问题。AlexNet搭建模型架构在模型最初被提出时,其结构如下图所示:该模型主要有以下几个优点:(1)首次利用GPU进行网络加速训练;(2)使用了Relu激活函数(),而不是传统的sigmoid
原创
2020-11-26 17:18:49 ·
1571 阅读 ·
1 评论