我是机器学习领域的新手,但我想尝试用 Keras 实现一个简单的分类算法。不幸的是,我的数据集非常少,所以我想尝试将迁移学习应用到这个问题上;但是,我在网上找不到任何东西,所以我想了解哪些是寻找预训练神经网络的最佳地点。在这方面您有什么建议吗?哪个网站最适合了解如何启动机器学习项目?
在哪里可以找到用于迁移学习的预训练模型
机器算法验证
机器学习
分类
神经网络
迁移学习
2022-03-09 09:11:34
2个回答
Keras 本身提供了一些在 ImageNet 上预训练的成功的图像处理神经网络: https ://keras.io/applications/
其他深度学习库也提供了一些预训练模型,特别是:
- TensorFlow:https ://github.com/tensorflow/models
- 咖啡:https ://github.com/BVLC/caffe/wiki/Model-Zoo
- caffe2:https ://github.com/caffe2/caffe2/wiki/Model-Zoo
- pytorch:https ://github.com/Cadene/pretrained-models.pytorch
- 千层面:https ://github.com/Lasagne/Recipes
还可以在https://www.gradientzoo.com找到许多适用于各种平台的预训练模型。
此外,如果您对某些特定的网络架构感兴趣,作者有时会自己提供预训练模型,例如ResNeXt。
由于问题标题是通用的(而不是特定于计算机视觉),我也会给出与NLP相关的答案,以防它对偶然发现预训练向量嵌入的人有所帮助:
可以在这些链接上找到两个最流行的预训练向量嵌入:
- 手套- https://nlp.stanford.edu/projects/glove/
- TensorFlow 嵌入 - https://code.google.com/archive/p/word2vec/
还有一些不太受欢迎和/或更近期的:
其它你可能感兴趣的问题