问题背景:使用horizon进行模型转换,将pt文件转换成onnx文件
运行结果及报错:
ERROR: 346_split_215_HzQuantize dims only support<=4
使用转换校验时发生的错误,没找到问题出在哪儿~
环境:Ubuntu20.04
模型:目标检测yolov5s模型
你好,应该是有一个算子的通道数超限了,需要进行对应的调整。
为了更好地适配后处理代码,我们在ONNX模型导出前对Github代码做了如下修改 (代码参见:https://github.com/ultralytics/yolov5/blob/v2.0/models/yolo.py%EF%BC%89%EF%BC%9A
def forward(self, x):
# x = x.copy() # for profiling
z = [] # inference output
self.training |= self.export
for i in range(self.nl):
x[i] = self.m[i](x[i]) # conv
bs, _, ny, nx = x[i].shape # x(bs,255,20,20) to x(bs,3,20,20,85)
# x[i] = x[i].view(bs, self.na, self.no, ny, nx).permute(0, 1, 3, 4, 2).contiguous()
x[i] = x[i].permute(0, 2, 3, 1).contiguous()
你把报错复制出来发给我