用预训练好的模型进行迁移学习

卷积神经网络

一般卷积层,ReLU层,池化层可以写成一个套餐
用预训练好的模型进行迁移学习_第1张图片

用预训练好的模型进行迁移学习

用人家已经训练好的模型(torchvision.model),已经训练好的参数,进行初始化!

数据量越,用预训练的时候基本上全部冻结住,自己加全连接层,数据量,可以将预训练模型少冻结一些,也是自己加全连接层。

ResNet

“好不好无所谓,但是一定不能比原来差。”所以深度可以继续延申。

网络结构

用预训练好的模型进行迁移学习_第2张图片

Input:图片输入

ZeroPad:边缘填充

BatchNorm:将图片进行归一化,普通数据是上述图a这样,经过迭代从橙色虚线收敛到紫色的线,但是如果将数据平移到原点,并且再进一步通过变换拉到数据之间的相对差异性,那么就会更快收敛到另一条“紫色”的线。

用预训练好的模型进行迁移学习_第3张图片

红色I模块(跳过)Identity_block:这种情况是三层卷积的输出结果格式一样

蓝色C模块(格式统一)Convolution_block:这种情况是三层卷积的输出结果格式不一样,所以需要将格式统一,利用1*1的卷积。

你可能感兴趣的:(联邦学习,实践笔记,迁移学习,深度学习,cnn)