大家好，我试图用retrain建立新图后，如何使用C api进行图像识别。
#include "neural_network.h"
void free_buffer(void *data, size_t length) { free(data); }
/* Read file containing the graph of the neural network and initialize the
* TF_Buffer
*/
TF_Buffer *read_file(const char *file) {
FILE *f = fopen(file, "rb");
fseek(f, 0, SEEK_END);
long fsize = ftell(f);
fseek(f, 0, SEEK_SET);
void *data = malloc(fsize);
fread(data, fsize, 1, f);
fclose(f);
TF_Buffer *buf = TF_NewBuffer();
buf->data = data;
buf->length = fsize;
buf->data_deallocator = free_buffer;
return buf;
}
// import the graph in the main graph
void import_graph(TF_Graph *graph) {
TF_Buffer *graph_def = read_file("output_graph.pb");
TF_Status *status = TF_NewStatus();
TF_ImportGraphDefOptions *opts = TF_NewImportGraphDefOptions();
TF_GraphImportGraphDef(graph, graph_def, opts, status);
TF_DeleteImportGraphDefOptions(opts);
// check status
if (TF_GetCode(status) != TF_OK) {
fprintf(stderr, "ERROR: Unable to import graph %s", TF_Message(status));
return;
}
TF_DeleteStatus(status);
TF_DeleteBuffer(graph_def);
}
static void Deallocator(void *data, size_t length, void *arg) {}
void *neural_network_task(void *period) {
struct timespec t;
unsigned int data[EL_W * ACT_IMG_H];
ssize_t x;
ssize_t line;
BITMAP *buf = create_bitmap(EL_W, ACT_IMG_H);
TF_Graph *graph = TF_NewGraph();
TF_Buffer *graph_def = read_file("retrained_graph.pb");
TF_Status *status = TF_NewStatus();
TF_ImportGraphDefOptions *opts = TF_NewImportGraphDefOptions();
TF_GraphImportGraphDef(graph, graph_def, opts, status);
TF_DeleteImportGraphDefOptions(opts);
// check status
if (TF_GetCode(status) != TF_OK) {
fprintf(stderr, "ERROR: Unable to import graph %s", TF_Message(status));
return;
}
const int nb_in = EL_W * ACT_IMG_H * sizeof(unsigned int);
const int nb_out = 3 * sizeof(float);
int64_t in_dims[] = {1, EL_W, ACT_IMG_H, 4};
int64_t out_dims[] = {1, 3};
set_activation(&t, *(int *)period);
// import_graph(graph);
while (1) {
acquire_bitmap(image);
// For each line in bitmap
for (line = 0; line < buf->h; ++line)
for (x = 0; x < buf->w; ++x) {
data[line * image->h + x] =
((unsigned int *)image->line[line])[x];
}
release_bitmap(image);
TF_Output input = {
TF_GraphOperationByName(graph, "DecodeJPGInput:0"), 0};
TF_Tensor *input_values =
TF_NewTensor(TF_UINT32, in_dims, 4, data, nb_in, &Deallocator, 0);
TF_Output output = {TF_GraphOperationByName(graph, "final_result:0"),
0};
TF_Tensor *output_values =
TF_AllocateTensor(TF_FLOAT, out_dims, 2, nb_out);
TF_SessionOptions *sess_opts = TF_NewSessionOptions();
TF_Session *session = TF_NewSession(graph, sess_opts, status);
TF_SessionRun(session, NULL, &input, &input_values, 1, &output,
&output_values, 1, NULL, 0, NULL, status);
TF_Code c = TF_GetCode(status);
printf("%i\n", c);
wait_for_activation(&t, *(int *)period);
}
}
在调用会话运行函数后，程序崩溃了，我不明白为什么，从api看来，参数应该是正确的。你们中有些人看到了我没看到的东西？
问题可能是TF_GraphOperationByName，即使在retrain.py中，这些都是返回的名称，但它们返回NULL。
编辑：
#include "neural_network.h"
#define GRAPH_NAME ("retrained_graph.pb")
#define IN_NAME ("Placeholder")
#define OUT_NAME ("final_result")
void free_buffer(void *data, size_t length) { free(data); }
/* Read file containing the graph of the neural network and initialize the
* TF_Buffer
*/
TF_Buffer *read_file(const char *file) {
FILE *f = fopen(file, "rb");
fseek(f, 0, SEEK_END);
long fsize = ftell(f);
fseek(f, 0, SEEK_SET);
void *data = malloc(fsize);
fread(data, fsize, 1, f);
fclose(f);
TF_Buffer *buf = TF_NewBuffer();
buf->data = data;
buf->length = fsize;
buf->data_deallocator = free_buffer;
return buf;
}
// import the graph in the main graph
void import_graph(TF_Graph *graph, TF_Status *status) {
TF_Buffer *graph_def = read_file(GRAPH_NAME);
TF_ImportGraphDefOptions *opts = TF_NewImportGraphDefOptions();
TF_GraphImportGraphDef(graph, graph_def, opts, status);
TF_DeleteImportGraphDefOptions(opts);
// check status
if (TF_GetCode(status) != TF_OK) {
fprintf(stderr, "ERROR: Unable to import graph %s", TF_Message(status));
return;
}
TF_DeleteBuffer(graph_def);
}
static void Deallocator(void *data, size_t length, void *arg) {}
TF_Code run_session(TF_Graph *graph, TF_Status *status, unsigned short *data) {
float *result;
// Number of bytes of input
const int nb_in = EL_W * ACT_IMG_H * sizeof(unsigned short);
// Number of bytes of output
const int nb_out = 3 * sizeof(float);
// Input dimensions
int64_t in_dims[] = {1, EL_W, ACT_IMG_H};
int n_in_dims = sizeof(in_dims) / sizeof(int64_t);
// Output dimensions
int64_t out_dims[] = {1, 3};
int n_out_dims = sizeof(out_dims) / sizeof(int64_t);
TF_Output input_op = {TF_GraphOperationByName(graph, IN_NAME), 0};
TF_Tensor *input_tensor[] = {TF_NewTensor(TF_UINT16, in_dims, n_in_dims,
data, nb_in, &Deallocator, 0)};
TF_Output output = {TF_GraphOperationByName(graph, OUT_NAME), 0};
TF_Tensor *output_values[] = {
TF_AllocateTensor(TF_FLOAT, out_dims, n_out_dims, nb_out)};
TF_SessionOptions *sess_opts = TF_NewSessionOptions();
TF_Session *session = TF_NewSession(graph, sess_opts, status);
TF_SessionRun(session,
NULL, // Run options
&input_op, &input_tensor[0], 1, // in: tensor, values, number
&output, &output_values[0], 1, // out: tensor, vlaues, number
NULL, 0, // target operation, num targets
NULL, // metadata
status // outputs status
);
// result = (float *)TF_TensorData(output_values);
// printf("%f %f %f\n", result[0], result[1], result[2]);
return TF_GetCode(status);
}
void image_linearization(unsigned short *data) {
BITMAP *image;
acquire_screen();
image = create_sub_bitmap(screen, IMG_XT, IMG_YT, EL_W, ACT_IMG_H);
ssize_t x;
ssize_t line;
for (line = 0; line < image->h; ++line)
for (x = 0; x < image->w; ++x) {
data[line * image->h + x] = ((short *)image->line[line])[x];
}
destroy_bitmap(image);
release_screen();
}
void *neural_network_task(void *period) {
struct timespec t;
unsigned short data[EL_W * ACT_IMG_H];
TF_Graph *graph = TF_NewGraph();
TF_Status *status = TF_NewStatus();
import_graph(graph, status);
set_activation(&t, *(int *)period);
/*
TF_Operation *decode = TF_GraphOperationByName(graph, IN_NAME);
printf("P: %p\n", decode);
printf("NO: %i\n", TF_OperationNumOutputs(decode));
printf("NI: %i\n", TF_OperationNumInputs(decode));
decode = TF_GraphOperationByName(graph, OUT_NAME);
printf("P: %p\n", decode);
printf("NO: %i\n", TF_OperationNumOutputs(decode));
printf("NI: %i\n", TF_OperationNumInputs(decode));
*/
while (1) {
image_linearization(data);
printf("%i\n", run_session(graph, status, data));
wait_for_activation(&t, *(int *)period);
}
}