SergeyLev
commented
3 years ago Попытался конвертировать модель в .tflite uint8, но выскакивает ошибка. ValueError: Failed to parse the model: pybind11::init(): factory function returned nullptr.
Возможно версия tensorflow не подходит, сейчас использую 2.3.1. Которая версия была использована при обучении? Скрипт конвертации
def representative_data_gen():
dataset_path = 'c:/DNet/nomeroff-net/datasets/autoriaNumberplateOcrEu-2020-10-09/train/img'
dataset_list = tf.data.Dataset.list_files(dataset_path)
for i in range(100):
image = next(iter(dataset_list))
image = tf.io.read_file(image)
image = tf.io.decode_jpeg(image, channels=3)
image = tf.image.resize(image, [1024, 1024])
image = tf.cast(image / 255., tf.float32)
image = tf.expand_dims(image, 0)
yield [image]
converter = tf.lite.TFLiteConverter.from_saved_model(SAVED_MODEL_DIR)
converter.optimizations = [tf.lite.Optimize.DEFAULT]
converter.target_spec.supported_ops = [tf.lite.OpsSet.TFLITE_BUILTINS_INT8]
converter.inference_input_type = tf.uint8
converter.inference_output_type = tf.uint8
converter.representative_dataset = representative_data_gen
tflite_model = converter.convert()
from os.path import join
tflite_model_path = join(SAVED_MODEL_DIR, 'eu_ocr_model.tflite' )
with open(tflite_model_path, 'wb') as f:
f.write(tflite_model)
ApelSYN
Добрый день! Есть идея запустить OCR на Coral Edge TPU, чтобы GPU ксавьера не загружать. Сначала решил спросить была ли попытка уже и есть ли смысл!