CenterFace.pytorch icon indicating copy to clipboard operation
CenterFace.pytorch copied to clipboard

Error in tensorrt inference

Open PankajJ08 opened this issue 4 years ago • 3 comments

There is an error in post-processing of tensorrt output. I converted the model into onnx and then trt. Running on jetson nano, cuda 10, tensorrt 6, torch 1.2. File "demo_tensorrt.py", line 107, in detections = body_engine.run(rgb_img)[1] File "ML/CenterFace.pytorch/TensorRT/centernet_tensorrt_engine.py", line 58, in run predictions = self.postprocess(trt_output, meta) File "ML/CenterFace.pytorch/TensorRT/centernet_tensorrt_engine.py", line 201, in postprocess hm, wh, hps, reg, hm_hp, hp_offset = args[0]; meta = args[1] ValueError: not enough values to unpack (expected 6, got 4)

PankajJ08 avatar Apr 13 '20 10:04 PankajJ08

Hi @chenjun2hao

Can you please address this issue as I too am getting the same error.

Thanks, Harshall.

harshall28 avatar May 01 '20 11:05 harshall28

I am also getting the same issue with cuda 10.2 tensorrt 7 Does anyone solve this issue?

Harryqu123 avatar May 19 '20 02:05 Harryqu123

i face the same issue. Have any update ?

GilbertTam avatar Sep 17 '20 05:09 GilbertTam