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

Where to get models for TransferLearning based on topics

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

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

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

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

但是,它没有给出任何结果,例如关键字 "sport"、"food" 或 "gun".


那么,找到所需 "topic" 的预训练模型的好方法是什么?

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

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

您通常可以在 model-zoo 网站上找到许多模型:https://modelzoo.co/

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

如果您需要代码(可能需要预训练的权重):paperswithcode.com 是搜索的好地方。