神经网络架构搜索出的网络EfficientNet,根据官方开源pytorch代码,转为caffe的网络协议文件,可以顺利训练,这里提供了EfficientNet B0的3种版本,第一种是完整 版,也就是使用了swish激活以及SE模块的,第二种是将swish替换为relu,且包含SE模块,第三种是将swish替换成relu,且去掉SE模块的最精简版本。经过在我自己的250类车辆分类数据集上,精简版可以达到0.9992的准确率,relu+SE版的准确率为0.9968,完整版本的还没训练结束,不过相差不多,效果很不错。
1