3.2使用PyTorch搭建AlexNet并训练花分类数据集

Posted 2021-05-30 NeverDelay

------------恢复内容开始------------

1、搭建AlexNet网络

2、如何使用自己的数据集使用网络

 

——————————————————————————————————————————————————————

1、搭建AlexNet网络

     网络结构：

 

 定义网络类：

• 定义初始化函数：定义网络在正向传播中所需要使用的层结构。

features:神经网络层

使用nn.Sequential模块：网络层次比较多可以使用nn.Sequential简洁代码

kernel_size:卷积核个数，stride:卷积核步长，padding：边缘补0的个数，如果是padding=(1,2)（上下,左右）则为上下方各补一行0，左右两侧各补两列零;如果上下左右补零行数都不同，则使用nn.ZeroPad2d((1,2,1,2))(左右上下)

classifier：全连接层

　　使用nn.Sequential模块

　　使用nn.Dropout

　　初始化判断init_weights（nn.Sequential中的参数）

• 定义前向传播

使用torch.flatten(x,start_dim=1):展平处理，索引从1开始

• 定义初始化判断函数