从哪里获取基于主题的迁移学习模型

Question

假设您正在搜索一个预训练模型，例如人类性别识别或年龄估计（迁移学习）。 因此，您需要一个网络训练，理想情况下是人脸，而不是 ImageNet 数据集之类的东西。

我知道搜索有两个很大的起点：

• Keras 应用程序
• TensorHub

现在，我发现最好的方法是使用 TensorHub 网站的搜索工具，例如 here。

这给了我一些在 CelebA-HQ 数据集上训练的模型，这是我一直在寻找的东西。

但是，它没有给出任何结果，例如关键字“运动”、“食物”或“枪”。

---

那么，为所需“主题”找到预训练模型的好方法是什么？

Answer 1

很难在一个地方为每个主题找到一个模型。

一般策略是在 GitHub 中搜索相关标签 ["tensorflow", "sport"]。

你可以在model-zoo网站上找到很多模型：https://modelzoo.co/

这也很有用：https://github.com/tensorflow/models

如果您需要代码（可能带有预先训练的权重）：paperswithcode.com 是一个搜索的好地方。