woshing700
commented
2 months ago 我按照以上思路做了，但是模型推理得到的结果不尽人意，不知道是不是我的数据读入方式有误
`
{"class": "0", "x": 100, "y": 65, "w": 6, "h": 5, "confidence": 19},
{"class": "0", "x": 31, "y": 36, "w": 10, "h": 8, "confidence": 30},
{"class": "0", "x": 17, "y": 66, "w": 8, "h": 4, "confidence": 12},
`
- 以下是我转换hpp的代码 `import cv2 import numpy import argparse
if name == 'main': parser = argparse.ArgumentParser(description='Model generator tool') parser.add_argument('-i', '--input', help='path to image') parser.add_argument('-o', '--output', help='path to image.hpp') args = parser.parse_args()
if args.input is None or args.output is None:
parser.print_help()
quit()
image = cv2.imread(args.input)
h,w,c=image.shape
with open(args.output, 'w') as file:
file.write('#pragma once\n'
'#include <stdint.h>\n\n'
f'#define IMAGE_HEIGHT {h}\n'
f'#define IMAGE_WIDTH {w}\n'
f'#define IMAGE_CHANNEL {c}\n\n'
'const static uint8_t image_data[] = {\n')
image = numpy.reshape(image, (-1,))
for i, element in enumerate(image[:-1], 1):
if i == 1:
file.write(' ')
file.write(f'{element}, ')
if i % 32 == 0:
file.write('\n ')
file.write(f'{image[-1]}')
file.write('};\n')`
LynnL4
sscma-example-esp32/components/modules/algorithm/algo_yolo.cpp
我在algo_yolo.cpp中找到了调用模型推理的代码，但不知如何改动；以下是一小部分代码
// Get information about the memory area to use for the model's input.input = interpreter->input(0);camera_fb_t *frame = NULL;uint16_t h = input->dims->data[1]; uint16_t w = input->dims->data[2]; uint16_t c = input->dims->data[3];if (c == 1) rgb565_to_gray(input->data.uint8, frame->buf, frame->height, frame->width, h, w, ROTATION_UP);else if (c == 3) rgb565_to_rgb888(input->data.uint8, frame->buf, frame->height, frame->width, h, w, ROTATION_UP);不知这个rgb565_to_rgb888()函数是否包含reszie的过程。