TF模型园和tf.keras.applications有什么区别?

时间:2020-01-09 11:50:41

标签: tensorflow keras

有了新的TensorFlow 2,我们有了他们的Model Garden(在GitHub的/models下),以及在tf.keras.applications下的Keras的预训练模型。

这两者有什么区别?

1 个答案:

答案 0 :(得分:1)

Tensorflow Keras应用程序
-CNN的预训练模型。它们包括最常用的CNN架构,例如ResNet,InceptionNet,VGG等。
-tf.keras.applications允许您直接导入CNN架构(请参阅文档https://www.tensorflow.org/api_docs/python/tf/keras/applications?hl=en)自定义其丢失并随心所欲地使用它。

Tensorflow模型花园
-用于其他任务的模型,例如NLP,对象检测等。这些模型也由TensorFlow维护。
-tf模型花园不是tf核心API的一部分,因此您不能像对tf.applications那样直接导入它们。