MATLAB中使用AlexNet、VGG、GoogLeNet进行迁移学习

最新推荐文章于 2023-03-03 17:06:39 发布

脉望虫

最新推荐文章于 2023-03-03 17:06:39 发布

阅读量5.2k

点赞数 3

CC 4.0 BY-SA版权

分类专栏： matlab 文章标签： matlab 深度学习神经网络

本文链接：https://blog.csdn.net/qq_36763031/article/details/105311334

直接贴代码，具体用法见注释：

clc;clear;

net = alexnet; %加载在ImageNet上预训练的网络模型
imageInputSize = [227 227 3];
%加载图像
allImages = imageDatastore('.\data227Alexnet',...
    'IncludeSubfolders',true,...
    'LabelSource','foldernames');
    %划分训练集和验证集
 [training_set,validation_set] = splitEachLabel(allImages,0.7,'randomized');
 %由于原始网络全连接层1000个输出，显然不适用于我们的分类任务，因此在这里替换
layersTransfer = net.Layers