search

NVIDIA-AI-IOT / tf_trt_models

TensorFlow models accelerated with NVIDIA TensorRT
BSD 3-Clause "New" or "Revised" License
682 stars 245 forks source link

tensorflow.python.framework.errors_impl.InvalidArgumentError #11

Open HilmiK opened 5 years ago

HilmiK commented 5 years ago

When I try faster_rcnn_resnet_101, when freezing the network I got this error from model_test.py which I make all steps in it :

File "model_test.py", line 37, in minimum_segment_size=50 File "/usr/local/lib/python2.7/dist-packages/tensorflow/contrib/tensorrt/python/trt_convert.py", line 115, in create_inference_graph int(msg[0])) tensorflow.python.framework.errors_impl.InvalidArgumentError: Invalid graph: Frame ids for node BatchMultiClassNonMaxSuppression/map/while/Reshape_1 does not match frame ids for it's fanout.

Thanks in advance.

ALEX-FISH commented 5 years ago

I have the same issue, any solution yet?

ghost commented 5 years ago

The solutions in this repository have only been tested to work against the SSD model variants. In the following issue use of Faster RCNN is discussed

https://github.com/NVIDIA-Jetson/tf_trt_models/issues/6#issuecomment-423098067