Re-trian SSD_incepiton_v2 can not work for sampleUffSSD.
Details on the platforms:
GPU type: GTX 1080
nvidia driver version: nvidia-418
CUDA version: CUDA10.0
Tensorflow version: 1.13.1
TensorRT version: 18.104.22.168
Describe the problem
I tried to convert Tensorflow frozen graph to uff file and ran inference with TensorRt.
It worked if I use the released official model (ssd_inception_v2_coco_2018_01_28), but it failed if I re-train the official model with Tensorflow-1.13.1 (same error with tensorflow 1.12)
./sample_uff_ssd ../data/ssd/sample_ssd_relu6.uff Begin parsing model... ERROR: UFFParser: Parser error: BoxPredictor_0/Reshape: Reshape: -1 dimension specified more than 1 time ERROR: sample_uff_ssd: Fail to parse sample_uff_ssd: sampleUffSSD.cpp:540: int main(int, char**): Assertion `tmpEngine != nullptr' failed. Aborted
Can anyone help? Thanks!