scrfd_onnx_tensorrt icon indicating copy to clipboard operation
scrfd_onnx_tensorrt copied to clipboard

Thank you

Open Msameim181 opened this issue 2 years ago • 0 comments

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     

Msameim181 avatar Oct 10 '22 13:10 Msameim181