Machine-Learning
在哪裡可以找到用於遷移學習的預訓練模型 [關閉]
我是機器學習領域的新手,但我想嘗試用 Keras 實現一個簡單的分類算法。不幸的是,我的數據集非常少,所以我想嘗試將遷移學習應用到這個問題上;但是,我在網上找不到任何東西,所以我想了解哪些是尋找預訓練神經網絡的最佳地點。在這方面您有什麼建議嗎?哪個網站最適合了解如何啟動機器學習項目?
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。