赞
踩
yolov5训练得到的pt模型转rknn模型,测试结果出错:出现重复框选,选框混乱的情况,如下图所示。
首先检查确认模型pt转onnx、onnx转rknn的整个流程步骤没出错,是按照教程来的:
# 原来的forward代码 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() if not self.training: # inference if self.grid[i].shape[2:4] != x[i].shape[2:4]: self.grid[i] = self._make_grid(nx, ny).to(x[i].device) y = x[i].sigmoid() y[..., 0:2] = (y[..., 0:2] * 2. - 0.5 + self.grid[i].to(x[i].device)) * self.stride[i] # xy y[...
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。