tensorflow/models

这是 TensorFlow 官方维护的模型库，相当于 TensorFlow 版的「model zoo」，收录了大量 SOTA 模型的参考实现，比如目标检测、图像分类、NLP 等任务的现成代码。它分几块：official 目录用最新的 TensorFlow 2 高层 API 实现并由官方持续更新，research 目录收录研究者贡献的模型，community 目录则是精选的第三方实现清单，另外还有一个叫 orbit 的轻量训练循环库，方便自定义训练逻辑并兼容 CPU、GPU、TPU。对从业者来说，它的价值在于当你需要复现某个论文的模型、或想参考官方推荐的建模最佳实践时，可以直接拿来对照，省去自己从零搭训练流程的功夫，部分模型还附带 TensorBoard 训练日志便于对比效果。适合正在用 TensorFlow 做研究或产品开发的算法工程师和研究员；如果你主要用 PyTorch 或者不做模型训练，这个仓库和你关系不大。

Star 趋势

仓库信息