EfficientNet_classification。EfficientNet在pytorch框架下实现图像分类,拿走即用。该文件包含python语言编写的model文件、my_dataset文件、predict文件、train文件、配置文件等。能够实现训练自己的数据集进行图像分类,以及对训练后的网络进行测试。EfficientNet利用NAS(Neural Architecture Search)搜索技术,将输入分辨率,网络的深度、宽度三者同时考虑,搭建更nice的网络结构。EfficientNet-B0的网络框架,总体看,分成了9个Stage:Stage1 是一个卷积核大小为3x3,步距为2的普通卷积层(包含BN和激活函数Swish);Stage2~Stage8 是在重复堆叠 MBConv 结构;Stage9 是一个普通的1x1的卷积层(包含BN和激活函数Swish) + 一个平均池化层 + 一个全连接层组成
1