Closed ycdhqzhiai closed 3 years ago
转onnx时候,修改了下Detect层
x[i] = x[i].view(bs, self.na, self.no, ny, nx).permute(0, 1, 3, 4, 2).contiguous()
改为
x[i] = torch.squeeze(x[i])
x[i] = x[i].view(self.na, self.no, ny, nx).permute(0, 2, 3, 1).contiguous()
回复原始代码就正常了,不大明白为什么
yolov5转onnx用的是原作者的么?
是的,自己加的squeeze转onnx是为了再其他平台上使用,去掉回复原始代码就好了
感谢,已经成功转成trt文件了,但是没有出框。我再仔细看看
@linghu8812 感谢开源,我在运行yolov5 simple时候报错
tensorRT新手,想问下这个错误是什么原因