对迁移学习进行网络外科手术
下面的例子演示了如何使用一套网络外科手术函数从预训练模型开始创建图像分类网络。
从 Wolfram Neural Net Repository 获取预先训练过的模型。
用 NetTake 提取网络的卷积部分。
用 NetJoin 添加新的分类层,同时附加一个新的 NetDecoder。
添加 ImageAugmentationLayer,用 NetPrepend 进行数据增强,并重新附加一个编码器。
定义数据集。
训练网络,保持预先训练的权重不变。
在未知输入上运行训练过的网络。