AttributeError: модуль 'tensorflow' не имеет атрибута 'lite' в модели Keras для преобразования Tensorflow Lite - Python

Я попытался преобразовать следующую модель keras в tflite для хостинга на мобильной платформе, используя следующий код. Я установил tenorflow version = 1.12 python version = 3.6.7 keras version = 2.2.4 Когда я запустил этот код, у меня возникла следующая ошибка.

convert = tf.lite.TFLiteConverter.from_keras_model_file (keras_file) AttributeError: модуль 'tensorflow' не имеет атрибута 'lite'

В чем может быть причина этой ошибки и как ее решить?

from keras.preprocessing.image import ImageDataGenerator
from keras.models import Sequential
from keras.layers import Conv2D, MaxPooling2D
from keras.layers import Activation, Dropout, Flatten, Dense
from keras import backend as K
import tensorflow as tf

# dimensions of our images.
img_width, img_height = 150, 150

train_data_dir = 'D:\\My Projects\\Dataset\\dataset6_2clz\\train'
validation_data_dir = 'D:\\My Projects\\Dataset\\dataset6_2clz\\validation'



nb_train_samples = 75
nb_validation_samples = 50
#epochs = 50
#batch_size = 16
epochs = 5
batch_size = 4

if K.image_data_format() == 'channels_first':
    input_shape = (3, img_width, img_height)
else:
    input_shape = (img_width, img_height, 3)

model = Sequential()
model.add(Conv2D(32, (3, 3), input_shape=input_shape))
model.add(Activation('relu'))
model.add(MaxPooling2D(pool_size=(2, 2)))

model.add(Conv2D(32, (3, 3)))
model.add(Activation('relu'))
model.add(MaxPooling2D(pool_size=(2, 2)))

model.add(Conv2D(64, (3, 3)))
model.add(Activation('relu'))
model.add(MaxPooling2D(pool_size=(2, 2)))

model.add(Flatten())
model.add(Dense(64))
model.add(Activation('relu'))
model.add(Dropout(0.5))
model.add(Dense(1))
model.add(Activation('sigmoid'))

model.compile(loss='binary_crossentropy',
              optimizer='rmsprop',
              metrics=['accuracy'])



train_datagen = ImageDataGenerator(
    rescale=1. / 255,
    shear_range=0.2,
    zoom_range=0.2,
    horizontal_flip=True)


test_datagen = ImageDataGenerator(rescale=1. / 255)

train_generator = train_datagen.flow_from_directory(
    train_data_dir,
    target_size=(img_width, img_height),
    batch_size=batch_size,
    class_mode='binary')

validation_generator = test_datagen.flow_from_directory(
    validation_data_dir,
    target_size=(img_width, img_height),
    batch_size=batch_size,
    class_mode='binary')

model.fit_generator(
    train_generator,
    steps_per_epoch=nb_train_samples // batch_size,
    epochs=epochs,
    validation_data=validation_generator,
    validation_steps=nb_validation_samples // batch_size)


# Save tf.keras model in HDF5 format.
keras_file = "7_try.h5"
model.save('7_try.h5')


# Convert to TensorFlow Lite model.

converter = tf.lite.TFLiteConverter.from_keras_model_file(keras_file)
tflite_model = converter.convert()
open("converted_model.tflite", "wb").write(tflite_model)

person Lakwin Chandula    schedule 08.12.2018    source источник
comment
Возможно ли, что ваш файл или любой другой файл в той же папке называется tensorflow.py?   -  person buran    schedule 08.12.2018
comment
@buran нет файла с именем tensorflow.py   -  person Lakwin Chandula    schedule 08.12.2018


Ответы (1)


В тензорном потоке 1.12 это должно быть converter = tf.contrib.lite.TFLiteConverter.from_keras_model_file(keras_file)

см. https://www.tensorflow.org/lite/convert/python_api#pre_tensorflow_1.12 Думаю, вы читали следующую ссылку https://www.tensorflow.org/lite/convert/python_api, но обратите внимание на следующее примечание

Примечание. В этих документах описывается конвертер в ночной версии TensorFlow, установленный с помощью pip install tf-nightly. Документы с описанием более старых версий см. В разделе «Конвертация моделей из TensorFlow 1.12».

Кроме того, для получения дополнительной информации вы можете увидеть это сообщение фиксации https://github.com/tensorflow/tensorflow/commit/61c6c84964b4aec80aeace187aab8cb2c3e55a72

person buran    schedule 08.12.2018