Как изменить архитектуру модели обнаружения объектов Tensorflow - PullRequest
0 голосов
/ 29 ноября 2018

Я пытаюсь реализовать тензор потока модель удаления объекта модель с movidius ncsdk2 , проблема в том, что ncsdk2 принимает только 1 выходной слой, а обнаружение объекта имеет 4 (класс, достоверность, поле,количество обнаружений), мне нужно каким-то образом суммировать эти 4 слоя всего за 1, это было возможно с помощью caffe framework, последний выходной слой содержит массив вроде этого

output[0] :  num_valid_boxes
output[base_index + 1] : prediction class index
output[base_index + 2] : prediction confidence
output[base_index + 3] : object boxpoint x1 value (it needs to be scaled)
output[base_index + 4] : object boxpoint y1 value (it needs to be scaled)
output[base_index + 5] : object boxpoint x2 value (it needs to be scaled)
output[base_index + 6] : object boxpoint y2 value (it needs to be scaled)

, где base_index увеличивается на 7 в каждомцикл, но я не знаю, с чего начать с tenorflow, я никогда не реализовывал свою собственную модель, примеры, которые я нашел в документации, очень просты по сравнению с моделью обнаружения объектов.

...