search

NVIDIA-AI-IOT / yolo_deepstream

yolo model qat and deploy with deepstream&tensorrt
Apache License 2.0
550 stars 139 forks source link

No module named 'trex' #35

Closed Levi-zhan closed 1 year ago

Levi-zhan commented 1 year ago

In draw-engine.py，it import trex, But this cannot be installed through pip。Is “trex“ a code?

wanghr323 commented 1 year ago

it is here. just clone the code https://github.com/NVIDIA/TensorRT/tree/main/tools/experimental/trt-engine-explorer

wanghr323 commented 1 year ago

we have already metion this on readme, Now add a notice on draw-engine.py too https://github.com/NVIDIA-AI-IOT/yolo_deepstream/commit/b36a83ba6177215383db358094ed875c975caa0c