search

NVIDIA-AI-IOT / tao-toolkit-triton-apps

Sample app code for deploying TAO Toolkit trained models to Triton
MIT License
84 stars 30 forks source link

ValueError: axes don't match array #6

Open ThiruRJST opened 2 years ago

ThiruRJST commented 2 years ago

The Logs:

File "tao_triton/python/entrypoints/tao_client.py", line 407, in <module>
    main()
  File "tao_triton/python/entrypoints/tao_client.py", line 399, in main
    postprocessor.apply(
  File "/home/tensorthug/ELabeller/tao_triton/python/postprocessing/detectnet_processor.py", line 122, in apply
    output_array[output_name] = results.as_numpy(output_name).transpose(0, 1, 3, 2)
ValueError: axes don't match array