Как запустить yolov2-lite tflite в USB-ускорителе Coral Edge TPU?

Я хотел бы убедиться, верны ли следующие шаги, которые я выполнил для получения tflite модели yolov2-lite?

Шаг 1 Сохранение графика и весов в файл protobuf
flow --model cfg/yolov2-tiny.cfg --load bin/yolov2-tiny.weights --savepb.
Эта команда создала папку build_graph с yolov2-tiny.pb и yolov2-tiny.meta.

Шаг 2 Преобразование 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 в ускорителе Coral Edge TPU USB.

Большое спасибо :)


person Abhi    schedule 04.11.2019    source источник


Ответы (1)


К сожалению, на данный момент модели yolo поддерживаются компилятором edgetpu. Рекомендую использовать модели mobile_ssd.

Для справки в будущем ваш конвейер должен быть:

1) Обучите модель

2) преобразовать в tflite

3) Скомпилировано для EdgeTPU (шаг, который фактически делегирует работу TPU)

person Nam Vu    schedule 05.11.2019
comment
Спасибо :). Кроме того, можно ли запускать модели tflite в Cloud TPU (я хотел бы сравнить производительность между ускорителем Edge coral TPU и Cloud TPUv2)? Еще раз спасибо. - person Abhi; 06.11.2019
comment
В этом я не уверен, хотя могу заверить вас, что облачный TPU v2 намного, намного быстрее lol Разница составляет 180 терафлопс против 4 терафлопс, но облачный tpu будет стоить несколько долларов / час - person Nam Vu; 06.11.2019
comment
Спасибо, Нам Ву. - person Abhi; 12.11.2019