diff --git a/models/cv/detection/yolox/ixrt/README.md b/models/cv/detection/yolox/ixrt/README.md index 102021d7cf04b035bb10424c7eaa37da9b39bd7d..79ac120530ee9213cd7be5f5939bf2c3eab649da 100644 --- a/models/cv/detection/yolox/ixrt/README.md +++ b/models/cv/detection/yolox/ixrt/README.md @@ -19,7 +19,8 @@ pip3 install onnx pip3 install onnxsim pip3 install tabulate pip3 install pycocotools -pip3 intsall ppq +pip3 install ppq +pip3 install pycuda pip3 install protobuf==3.20.0 ``` @@ -50,6 +51,9 @@ export DATASETS_DIR=/Path/to/coco/ ```bash # build plugin cd plugin && mkdir build && cd build +## link libixrt.so to lib +ln -s /usr/local/corex/lib/python3/dist-packages/tensorrt/lib/libixrt.so /usr/local/corex/lib +## make cmake .. -DIXRT_HOME=/usr/local/corex make -j12 # Accuracy @@ -99,4 +103,4 @@ yolox | 32 | INT8 | 832.16 | 0.647 | ## Reference -YOLOX: https://github.com/Megvii-BaseDetection/YOLOX \ No newline at end of file +YOLOX: https://github.com/Megvii-BaseDetection/YOLOX diff --git a/models/cv/detection/yolox/ixrt/plugin/cmake/FindIxrt.cmake b/models/cv/detection/yolox/ixrt/plugin/cmake/FindIxrt.cmake index 5b0f27293edaebf80cd5bfd622c363f49b36966b..f80c45d93f444d1dee9ecbb4ef37130b6d4d83fc 100644 --- a/models/cv/detection/yolox/ixrt/plugin/cmake/FindIxrt.cmake +++ b/models/cv/detection/yolox/ixrt/plugin/cmake/FindIxrt.cmake @@ -3,6 +3,7 @@ if(NOT "${IXRT_HOME}" STREQUAL "") set(IXRT_INCLUDE_DIR ${IXRT_HOME}/include) set(IXRT_LIB_DIR ${IXRT_HOME}/lib) + include_directories("${IXRT_HOME}/lib/python3/dist-packages/tensorrt/include") # From default paths else() set(IXRT_INCLUDE_DIR /usr/local/corex/include) @@ -16,4 +17,4 @@ if(EXISTS ${IXRT_INCLUDE_DIR} AND EXISTS ${IXRT_LIB_DIR}) include_directories(${IXRT_INCLUDE_DIR}) else() message( FATAL_ERROR "IxRT library doesn't exist!") -endif() \ No newline at end of file +endif()