PyTorch实现EfficientNet图像分类模型

EfficientNet_classification。基于PyTorch框架下的EfficientNet图像分类模型实现，包含Python编写的model、dataset、predict、train等文件。该模型利用了NAS搜索技术，兼顾输入分辨率、深度、宽度等因素，搭建了性能更优秀的网络结构。通过训练自己的数据集可进行图像分类，测试也十分方便。本文详细介绍了EfficientNet-B0的网络框架，阐述了各个Stage的作用，让读者对该模型有更深入的了解。