本地训练模型是否与老平台训练模型输出不同

Viewed 28

使用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值与标签值均已修改,似乎是模型输出不同的问题,请问我该如何进行修改呢

1 Answers

当然,不要把它当一套东西,你用训练平台就全部按照训练平台的那一套东西走,用YOLO就全按照YOLO那一套东西走,不要互相交换模型代码。