I change the fixed_shape_resizer of ssd_indeption_v2 from 300x300 into 600x600 and trained my dataset. When I use model to detected image with the size 1280x720 in tensorrt, it’s return the following result:
Begin parsing model…
End parsing model…
Begin building engine…
End building engine…
Num batches 2
Data Size 5529600
input_c: 3 input_h: 1280 input_w: 720
Cuda failure: 11Aborted (core dumped)
I have changed the INPUT_H and INPUT_W value in BatBatchStreamPPM.h file. If I detect the image smaller than 600x600, there is no error. Does this mean that the size of the picture should not exceed 600x600?
The version of tensorrt I use is TensorRT-188.8.131.52.
Can anyone help me?