Я хотел бы убедиться, что следующие шаги, которые я выполнил для получения tflite модели yolov2-lite, верны или нет?
Step1 Сохранение графика и весов в protobuffile
flow --model cfg/yolov2-tiny.cfg --load bin/yolov2-tiny.weights --savepb.
Эта команда создала папку build_graph с yolov2-tiny.pb и yolov2-tiny.meta.
Step2 Преобразование pb в tflite
Я выполнил приведенный ниже фрагмент кода, чтобы получить yolov2-tiny.tflite
import tensorflow as tf
localpb = 'yolov2-tiny.pb'
tflite_file = 'yolov2-tiny.tflite'
print("{} -> {}".format(localpb, tflite_file))
converter = tf.lite.TFLiteConverter.from_frozen_graph(
localpb,
input_arrays= ['input'],
output_arrays= ['output']
)
tflite_model = converter.convert()
open(tflite_file,'wb').write(tflite_model)
interpreter = tf.lite.Interpreter(model_content=tflite_model)
interpreter.allocate_tensors()
Если приведенные выше шаги, которые я выполнил, чтобы получить этот tflite, верны, то, пожалуйста, предложите мне команду для запуска этого файла tflite вКоралловый край ТПУ USB-ускоритель.
Большое вам спасибо:)