从哪里获得基于主题的迁移学习模型
Where to get models for TransferLearning based on topics
假设您正在搜索预训练模型,例如人类性别识别或年龄估计(迁移学习)。
因此,您需要一个经过训练的网络,理想情况下是人脸,而不是像 ImageNet 数据集这样的东西。
我知道搜索有两个大的起点:
- Keras 应用程序
- TensorHub
现在,我发现最好的方法是使用 TensorHub 网站的搜索工具,例如 here。
这给了我一些在 CelebA-HQ 数据集上训练的模型,这是我一直在寻找的东西。
但是,它没有给出任何结果,例如关键字 "sport"、"food" 或 "gun".
那么,找到所需 "topic" 的预训练模型的好方法是什么?
很难在一个地方找到每个主题的模型。
一般策略可以在 GitHub 中搜索相关标签 ["tensorflow", "sport"]
。
您通常可以在 model-zoo 网站上找到许多模型:https://modelzoo.co/
这也很有用:https://github.com/tensorflow/models
如果您需要代码(可能需要预训练的权重):paperswithcode.com 是搜索的好地方。
假设您正在搜索预训练模型,例如人类性别识别或年龄估计(迁移学习)。 因此,您需要一个经过训练的网络,理想情况下是人脸,而不是像 ImageNet 数据集这样的东西。
我知道搜索有两个大的起点:
- Keras 应用程序
- TensorHub
现在,我发现最好的方法是使用 TensorHub 网站的搜索工具,例如 here。
这给了我一些在 CelebA-HQ 数据集上训练的模型,这是我一直在寻找的东西。
但是,它没有给出任何结果,例如关键字 "sport"、"food" 或 "gun".
那么,找到所需 "topic" 的预训练模型的好方法是什么?
很难在一个地方找到每个主题的模型。
一般策略可以在 GitHub 中搜索相关标签 ["tensorflow", "sport"]
。
您通常可以在 model-zoo 网站上找到许多模型:https://modelzoo.co/
这也很有用:https://github.com/tensorflow/models
如果您需要代码(可能需要预训练的权重):paperswithcode.com 是搜索的好地方。