efficientnet

收录理由

EfficientNet 是一种轻量级卷积网络，在 ImageNet 上达到顶尖精度的同时，参数量和计算量却比同类模型少一个数量级。这个 Keras 复现版本提供了 B0 到 B7 全系列预训练权重，你可以根据实际场景选择：推理速度或内存紧张时用小型号，追求精度时换大型号。权重覆盖 ImageNet 和另外五个常用的迁移学习数据集，拿来微调自定义分类器相当省事。代码高度贴近原始的 TensorFlow 实现，如果你不想把它当黑盒，而是想研究或改动网络结构，这份代码也是很好的参考。