EfficientNetCaffe.zip

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