Carnon / trt_yolov7

tensorrt yolov7 without onnxparser
24 stars 3 forks source link

study yolov7

without onnx parser

ubuntu18.04 cuda-11.1 tensorrt-7.2.2.3 opencv

    git clone https://github.com/WongKinYiu/yolov7.git

    cp {trt_yolov7}/gen_wts.py {WongKinYiu}/yolov7

    cd {WongKinYiu}/yolov7

    wget https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7-w6-pose.pt

    python gen_wts.py

    cp {WongKinYiu}/yolov7-w6-pose.wts {trt_yolov7}/

    cd {trt_yolov7}

    vim cmakelists.txt   and  set tensort path; this depend opencv cuda tensorrt

    mkdir build

    cd build

    cmake ..

    make

    ./convert -s ../yolov7-w6-pose.wts ../yolov7-w6-pose.engine

    python inference.py