scrfd_onnx_tensorrt
scrfd_onnx_tensorrt copied to clipboard
Thank you
Thank you very much for your hint.
This needs a little adjustment: out[np.newaxis]
trt_outs = do_inference(context=self.context, bindings=self.bindings, inputs=self.inputs,
outputs=self.outputs, stream=self.stream)
net_outs = []
reshape_list = [1, 4, 10] # reshape 第二位的数
k = 0
i = 0
while len(net_outs) < 9:
out = trt_outs[i]
out = out.reshape(out.shape[0]//reshape_list[k],reshape_list[k])
net_outs.append(out[np.newaxis])
i += 3
if i >8:
k += 1
i = k