Тензор потока в Keras: ошибка импорта графика в модель Keras

Anna Jeanine спросил: 03 ноября 2018 в 08:13 в: python

У меня есть код Tensorflow для классификации изображений, который я хочу преобразовать в код Keras. Но у меня проблемы с API более высокого уровня, не имеющего все коды, которые я желаю. Проблема, на которой я застрял, состоит в следующем:

#net = get_vgg_model() <- got tf.VGG16 model
net = tf.keras.applications.VGG16()g1 = tf.Graph()
with tf.Session(graph=g1, config=config) as sess, g1.device('/cpu:0'):
    tf.import_graph_def(net['graph_def'], name='vgg')

этот код выдает ошибку:

Traceback (most recent call last):
  File "app.py", line 16, in <module>
    from modules.xvision import Xvision
    File "/app/modules/xvision.py", line 84, in <module>
       tf.import_graph_def(net['graph_def'], name='vgg')
   TypeError: 'Model' object has no attribute '__getitem__'

Может ли кто-нибудь помочь мне с этим график?


0 ответов