Thanks to visit codestin.com
Credit goes to github.com

Skip to content

tensorrt yolov7 without onnxparser

Carnon/trt_yolov7

Repository files navigation

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

About

tensorrt yolov7 without onnxparser

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published