Как интегрировать модель Vgg, созданную в Keras, в приложение C? У меня есть файл (.h5)Python

Программы на Python
Ответить Пред. темаСлед. тема
Anonymous
 Как интегрировать модель Vgg, созданную в Keras, в приложение C? У меня есть файл (.h5)

Сообщение Anonymous »

Я работаю над исследованием, в ходе которого разработал модель Vgg с использованием Keras. Но мне нужно реализовать эту модель в микроконтроллере, и поэтому я подумывал об интеграции этой модели в вызов приложения C.
Изначально я думал об использовании C API tensorflow. Для этого я преобразовал файл (.h5) в (.pb) — просто чтобы сделать его совместимым с найденным примером:
Код Python для .h5 > . конверсия пб

Код: Выделить всё

import tensorflow as tf
model = tf.keras.models.load_model('path_to_your_model.h5')
tf.saved_model.save(model, 'saved_model_directory')
Код для вызова модели Keras на C

Код: Выделить всё

#include 

int main() {
// Initialize TensorFlow
TF_Status* status = TF_NewStatus();
TF_SessionOptions* options = TF_NewSessionOptions();
TF_Session* session = TF_NewSession(options, status);
TF_Buffer* run_options = NULL;
const char* tags = "serve";  // Use "serve" tag for inference

// Load the SavedModel
TF_Graph* graph = TF_NewGraph();
TF_Buffer* saved_model_buffer = TF_ReadFile("path_to_saved_model_directory", status);
TF_SessionRun(session, run_options,
NULL, NULL, 0,
NULL, NULL, 0,
NULL, 0,
&graph, status);
TF_DeleteBuffer(saved_model_buffer);

// Perform inference
// Construct input tensor, run session, get output tensor, etc.

// Clean up
TF_DeleteGraph(graph);
TF_DeleteSession(session, status);
TF_DeleteSessionOptions(options);
TF_DeleteStatus(status);

return 0;
}
Может ли кто-нибудь подтвердить, имеет ли эта реализация смысл? Потому что, когда я запускаю следующий bash, g++ -IC:/Users/dossa/Desktop/applicationC/include main.c -LC:/Users/dossa/Desktop/applicationC/lib -ltensorflow -o integracao, я получаю ошибка:

Код: Выделить всё

C:\Users\dossa\AppData\Local\Temp\ccTGNpqh.o:main.c:(.text+0xf): undefined reference to `_imp__TF_NewStatus'
C:\Users\dossa\AppData\Local\Temp\ccTGNpqh.o:main.c:(.text+0x1a): undefined reference to `_imp__TF_NewSessionOptions'
C:\Users\dossa\AppData\Local\Temp\ccTGNpqh.o:main.c:(.text+0x25): undefined reference to `_imp__TF_NewGraph'
C:\Users\dossa\AppData\Local\Temp\ccTGNpqh.o:main.c:(.text+0xd8): undefined reference to `_imp__TF_NewBuffer'
C:\Users\dossa\AppData\Local\Temp\ccTGNpqh.o:main.c:(.text+0x103): undefined reference to `_imp__TF_NewImportGraphDefOptions'
C:\Users\dossa\AppData\Local\Temp\ccTGNpqh.o:main.c:(.text+0x12d): undefined reference to `_imp__TF_GraphImportGraphDef'C:\Users\dossa\AppData\Local\Temp\ccTGNpqh.o:main.c:(.text+0x13b): undefined reference to `_imp__TF_DeleteImportGraphDefOptions'
C:\Users\dossa\AppData\Local\Temp\ccTGNpqh.o:main.c:(.text+0x159): undefined reference to `_imp__TF_NewSession'
C:\Users\dossa\AppData\Local\Temp\ccTGNpqh.o:main.c:(.text+0x16b): undefined reference to `_imp__TF_GetCode'
C:\Users\dossa\AppData\Local\Temp\ccTGNpqh.o:main.c:(.text+0x182): undefined reference to `_imp__TF_Message'
C:\Users\dossa\AppData\Local\Temp\ccTGNpqh.o:main.c:(.text+0x1a7): undefined reference to `_imp__TF_DeleteBuffer'
C:\Users\dossa\AppData\Local\Temp\ccTGNpqh.o:main.c:(.text+0x1b5): undefined reference to `_imp__TF_DeleteSessionOptions'
C:\Users\dossa\AppData\Local\Temp\ccTGNpqh.o:main.c:(.text+0x1cb): undefined reference to `_imp__TF_DeleteSession'
C:\Users\dossa\AppData\Local\Temp\ccTGNpqh.o:main.c:(.text+0x1d9): undefined reference to `_imp__TF_DeleteGraph'
C:\Users\dossa\AppData\Local\Temp\ccTGNpqh.o:main.c:(.text+0x1e7): undefined reference to `_imp__TF_DeleteStatus'
collect2.exe: error: ld returned 1 exit status
Что я должен получить?

Подробнее здесь: https://stackoverflow.com/questions/783 ... ave-a-file
Реклама
Ответить Пред. темаСлед. тема

Быстрый ответ

Изменение регистра текста: 
Смайлики
:) :( :oops: :roll: :wink: :muza: :clever: :sorry: :angel: :read: *x)
Ещё смайлики…
   
К этому ответу прикреплено по крайней мере одно вложение.

Если вы не хотите добавлять вложения, оставьте поля пустыми.

Максимально разрешённый размер вложения: 15 МБ.

  • Похожие темы
    Ответы
    Просмотры
    Последнее сообщение

Вернуться в «Python»