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