#642 · 主分类: 计算机视觉

cnn_finetune

在Keras中微调CNN

项目最后更新:08/11/18

GitHub Stars

940

Forks数量

402

贡献者数量

1

许可证

MIT

收录理由

在 Keras 里做迁移学习的人大概都体会过这种麻烦:想找个现成的 ImageNet 预训练权重,还得自己琢磨预处理流程。这个仓库把 VGG、Inception、ResNet、DenseNet 的微调示例都整理成了独立脚本,每个脚本都附带了对应的预训练权重下载链接。在 Cifar10 上跑一个脚本,就能完整看到整个流程:加载模型、替换分类头、重新训练。它还特别处理了最容易出错的预处理细节,比如把 RGB 转成 BGR、减去 ImageNet 均值像素。无论是想学习迁移学习,还是打算把参考实现改成自己的数据集,这里都是一个清晰、具体的起点。

相关文章

暂无针对该项目的文章。

如需补充新闻资讯、使用教程、应用测评等内容,欢迎 联系我们推荐选题或投稿。

同类热门项目