Re-trian SSD_incepiton_v2 can not work for sampleUffSSD.
Details on the platforms:
Linux: Ubuntu16.06
GPU type: GTX 1080
nvidia driver version: nvidia-418
CUDA version: CUDA10.0
Tensorflow version: 1.13.1
TensorRT version: 5.0.2.6
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!