search

cqu20160901 / yolov10_onnx_rknn_horizon_tensorRT

yolov10 目标检测部署版本，便于移植不同平台（onnx、tensorRT、rknn、Horizon），全网部署最简单、运行速度最快的部署方式（全网首发）。
47 stars 7 forks source link

模型输出顺序问题 #7

Open HalcyonWangSZ opened 5 hours ago

HalcyonWangSZ commented 5 hours ago

你好，我使用官方库训练的模型尝试推理时遇到了如下报错 cls_val = cls[cl map_size[index][0] map_size[index][1] + h * map_size[index][1] + w]


IndexError: index 409600 is out of bounds for axis 0 with size 409600
检查你的模型的out发现是reg1, reg2, reg3, cls1, cls2, cls3, seg, mc1, mc2, mc3，与最新的官方模型对不上，请问你是用哪一个版本的yolov8训练的，我尝试了几个版本都不对
cqu20160901 commented 5 hours ago

我用的是yolov10的官方代码训练的