79 lines
2.2 KiB
C
79 lines
2.2 KiB
C
|
/*
|
|||
|
|
* Licensed to the Apache Software Foundation (ASF) under one
|
|||
|
|
* or more contributor license agreements. See the NOTICE file
|
|||
|
|
* distributed with this work for additional information
|
|||
|
|
* regarding copyright ownership. The ASF licenses this file
|
|||
|
|
* to you under the Apache License, Version 2.0 (the
|
|||
|
|
* "License"); you may not use this file except in compliance
|
|||
|
|
* with the License. You may obtain a copy of the License at
|
|||
|
|
*
|
|||
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|||
|
|
*
|
|||
|
|
* Unless required by applicable law or agreed to in writing,
|
|||
|
|
* software distributed under the License is distributed on an
|
|||
|
|
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|||
|
|
* KIND, either express or implied. See the License for the
|
|||
|
|
* specific language governing permissions and limitations
|
|||
|
|
* under the License.
|
|||
|
|
*/
|
|||
|
|
|
|||
|
|
#include <stdio.h>
|
|||
|
|
#include <tvm_runtime.h>
|
|||
|
|
#include <tvmgen_rec.h>
|
|||
|
|
|
|||
|
|
#include "uart.h"
|
|||
|
|
|
|||
|
|
// Header files generated by convert_image.py
|
|||
|
|
#include "inputs.h"
|
|||
|
|
#include "outputs.h"
|
|||
|
|
|
|||
|
|
|
|||
|
|
int main(int argc, char** argv) {
|
|||
|
|
char dict[]={"#0123456789:;<=>?@ABCDEFGHIJKLMNOPQRSTUVWXYZ[\\]^_`abcdefghijklmnopqrstuvwxyz{|}~!\"#$%&'()*+,-./ "};
|
|||
|
|
int char_dict_nums = 97;
|
|||
|
|
uart_init();
|
|||
|
|
printf("Starting ocr rec inference\n");
|
|||
|
|
struct tvmgen_rec_outputs rec_outputs = {
|
|||
|
|
.output = output,
|
|||
|
|
};
|
|||
|
|
struct tvmgen_rec_inputs rec_inputs = {
|
|||
|
|
.x = input,
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
tvmgen_rec_run(&rec_inputs, &rec_outputs);
|
|||
|
|
|
|||
|
|
// post process
|
|||
|
|
int char_nums = output_len / char_dict_nums;
|
|||
|
|
|
|||
|
|
int last_index = 0;
|
|||
|
|
float score = 0.f;
|
|||
|
|
int count = 0;
|
|||
|
|
|
|||
|
|
printf("text: ");
|
|||
|
|
for (int i = 0; i < char_nums; i++) {
|
|||
|
|
int argmax_idx = 0;
|
|||
|
|
float max_value = 0.0f;
|
|||
|
|
for (int j = 0; j < char_dict_nums; j++){
|
|||
|
|
if (output[i * char_dict_nums + j] > max_value){
|
|||
|
|
max_value = output[i * char_dict_nums + j];
|
|||
|
|
argmax_idx = j;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
if (argmax_idx > 0 && (!(i > 0 && argmax_idx == last_index))) {
|
|||
|
|
score += max_value;
|
|||
|
|
count += 1;
|
|||
|
|
// printf("%d，%f，%c\n", argmax_idx, max_value, dict[argmax_idx]);
|
|||
|
|
printf("%c", dict[argmax_idx]);
|
|||
|
|
}
|
|||
|
|
last_index = argmax_idx;
|
|||
|
|
}
|
|||
|
|
score /= count;
|
|||
|
|
printf(", score: %f\n", score);
|
|||
|
|
|
|||
|
|
// The FVP will shut down when it receives "EXITTHESIM" on the UART
|
|||
|
|
printf("EXITTHESIM\n");
|
|||
|
|
while (1 == 1)
|
|||
|
|
;
|
|||
|
|
return 0;
|
|||
|
|
}
|