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

finetune_alexnet_with_tensorflow

alexnet deep-learning fine-tune tensorflow

用于在TensorFlow >= 1.2rc0中微调AlexNet的代码

项目最后更新:03/05/19

GitHub Stars

777

Forks数量

401

贡献者数量

2

许可证

BSD-3-Clause

收录理由

这个仓库把微调AlexNet这件事讲得明明白白,代码量不大,从头读到尾也不费劲。里面包含了完整的AlexNet网络结构定义、一个直接读取图片路径和类别标签文本文件就能跑的微调脚本,还有配合TensorFlow输入管道的数据生成器封装,另外附带了用ImageNet图片验证预训练权重的notebook。对于想搞懂迁移学习具体怎么接线的人来说,是个很实在的学习样本。虽然它用的TensorFlow接口和网络结构本身都偏老,放到现在未必是团队会优先选的技术方案,但作为理解微调流程的参考,价值依然在。

相关文章

暂无针对该项目的文章。

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

同类热门项目