yolov5_fastreid_deepsort_tensorrt icon indicating copy to clipboard operation
yolov5_fastreid_deepsort_tensorrt copied to clipboard

how pre_precessing run on gpu with tensorrt

Open gaokao123 opened this issue 2 years ago • 1 comments

in this demo:https://github.com/linghu8812/yolov5_fastreid_deepsort_tensorrt, the pre_processing is run on cpu with opencv, how to run on gpu with tensorrt, do you have any ideas or any demos? thank you

gaokao123 avatar Dec 26 '22 03:12 gaokao123

in this demo:https://github.com/linghu8812/yolov5_fastreid_deepsort_tensorrt, the pre_processing is run on cpu with opencv, how to run on gpu with tensorrt, do you have any ideas or any demos? thank you

export sub and devide op at the beggin of ONNX model.

linghu8812 avatar Jan 05 '23 10:01 linghu8812