iwatake2222 / play_with_tensorrt

Sample projects for TensorRT in C++
Apache License 2.0
191 stars 34 forks source link

how can I process my images before putting into hitnet predict ? #18

Closed Surtr07 closed 2 years ago

Surtr07 commented 2 years ago

I try the hitnet_with_trt trained by ETH3d,but the result is very bad .So I guess if my input image need to be any process before predicting?

iwatake2222 commented 2 years ago

I don't know your training code, but you need the same preprocess for inference as the training code.