Pycharm虚拟环境中的Tensorflow问题:TypeError:“模型”对象不可迭代

时间:2019-07-18 11:13:15

标签: tensorflow keras pycharm virtualenv

我在ubuntu中使用Python 3.5,Tensorflow-gpu 1.12.0,Keras 2.2.4。当我在Pycharm中使用系统解释器时,代码运行没有任何问题。但是,当我在Pycharm中创建虚拟环境并安装所有必需软件包(OpenCV,Sklearn,pandas,Keras,Tensorflow)的相同版本时,会出现以下错误:

Traceback (most recent call last):
  File "/media/ehsan/48BE4782BE476810/AA_MY_PYTHON_CODE/MultiLable_MultiTask_Light_Examples/CodeTwo/2_Main_Code_Training_Multitask_Network.py", line 338, in <module>
    base_model, multi_model, feature_map = multi_model(loss_list, test_metrics, dd)
  File "/media/ehsan/48BE4782BE476810/AA_MY_PYTHON_CODE/MultiLable_MultiTask_Light_Examples/CodeTwo/2_Main_Code_Training_Multitask_Network.py", line 40, in multi_model
    _, base_model = VGG19(weights='imagenet', include_top=False, input_shape=(175, 100, 3))
TypeError: 'Model' object is not iterable

我试图重新安装Tensorflow和Keras。另外,我重新创建了虚拟环境。但是,在使用虚拟环境时出现了同样的错误。

1 个答案:

答案 0 :(得分:0)

尝试从TensorFlow而不是Keras导入模型。

from tensorflow.keras.models import load_model

代替

from keras.models import load_model