使用yolov5s本地训练的kmodel模型
在运行如下代码时出现了报错
results = []
for i in range(kpu.outputs_size()):
out_data = kpu.get_output_tensor(i)
result = out_data.to_numpy()
result = result.reshape((result.shape[0]*result.shape[1]*result.shape[2]*result.shape[3]))
del out_data
results.append(result)
gc.collect()
报错信息如下:
File "<stdin>", line 86, in object_detection_task
IndexError: list index out of range
训练的模型与云平台相同,均为640*640,deploy文件中anchor值与标签值均已修改,似乎是模型输出不同的问题,请问我该如何进行修改呢