how pre_precessing run on gpu with tensorrt
gaokao123 opened this issue · 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
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.