tensorflow2沒有slim模塊

錯誤原因

在TensorFlow2的版本中,已經移除了contrib模塊,所以就沒有slim模塊,在TensorFlow1的版本中slim的模塊導入如下

from tensorflow.contrib import slim

解決辦法

我們可以通過安裝tf_slim模塊來替換contrib中slim,安裝tf_slim模塊

pip install tf_slim
import tf_slim as slim

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章