efficient作为一个2019年的新模型,思想就是用一个系数来改变模型深度、宽度与分辨率来提高性能
这篇文章主要讲如何快速应用efficient net,原理可以看原论文
或者
https://blog.youkuaiyun.com/BEYONDMA/article/details/91146289
这里我们可以首先直接使用别人的库
github 地址
Requirements
Keras >= 2.2.0 / TensorFlow >= 1.12.0
keras_applications >= 1.0.7
scikit-image
首先安装库
pip install -U efficientnet
注意项目中具有tensorflow1.x与tensorflow2.x的用法
1.x:
import efficientnet.keras as efn
model = efn.EfficientNetB0(weights='imagenet')
2.x
import efficientnet.tfkeras as efn
model = efn.EfficientNetB0(weights='imagenet')
载入权重:
import efficientnet.tfkeras