search

kunyao2015 / TensorRT_Yolo

2 stars 2 forks source link

help me thanks #1

Open shining-love opened 3 years ago

shining-love commented 3 years ago

hello， could you show me that how to get deepsort.engine?

kunyao2015 commented 3 years ago

I used this project : https://github.com/ZQPei/deep_sort_pytorch . Actually, the deep part is a person reid task and the network is simple 1) convert torch moel to onnx format, 2) then convert onnx file to TensorRT engine and serialize the engine to deepsort.engine(or any other names)

shining-love commented 3 years ago

I used this project : https://github.com/ZQPei/deep_sort_pytorch . Actually, the deep part is a person reid task and the network is simple

  1. convert torch moel to onnx format, 2) then convert onnx file to TensorRT engine and serialize the engine to deepsort.engine(or any other names)

Thanks for your reply! could you show me how to convert torch moel to onnx format, and then convert onnx file to TensorRT engine?any code?