请问如何python调用deepsort.engine文件？

cong / yolov5_deepsort_tensorrt_cpp

This repo is a C++ version of yolov5_deepsort_tensorrt. Packing all C++ programs into .so files, using Python script to call C++ programs further.

MIT License

49 stars 5 forks source link

请问如何python调用deepsort.engine文件？ #3

Open StawEndl opened 2 years ago

StawEndl commented 2 years ago

我现在是可以调用预测的engine文件了（根据你之前的加速模型预测那部分github的代码），但是deepsort部分的engine文件不会调用，是跟预测那部分一样吗？如果是，cuda.memcpy_htod_async(cuda_inputs[0], host_inputs[0], stream)中的host_inputs格式应该怎么写？因为模型预测只传图片，但是deepsort要传多个变量。。

cong commented 2 years ago

其实优化的还是 deepsort 特征提取部分