Alexnet实现Caltech101数据集图像分类(pytorch实现)
从101个文件中读取图像数据(进行resize和RGB的转化,原始图像数据大小不一,必须resize),并为其加上101类标签(0-100)进行图像变换,并分出训练集,验证集和测试集自定义一个数据集类(继承自dataset)便于数据加载网络结构Alexnet模型由5个卷积层和3个池化Pooling 层 ,其中还有3个全连接层构成。AlexNet 跟 LeNet 结构类似,但使⽤了更多的卷积层和更⼤的参数空间来拟合⼤规模数据集 ImageNet。它是浅层神经⽹络和深度神经⽹络的分界线。Alexnet
原创
2022-06-10 19:05:34 ·
6090 阅读 ·
5 评论