build ssd engine failed about tensorrt_demos

[jkjung-avt]

@Ndron Could you check and make sure you have patched "graphsurgeon/node_manipulation.py"? More specifically, you have to add the following line of code into the file:

+    node.attr["dtype"].type = 1

Reference:

• https://github.com/jkjung-avt/tensorrt_demos/blob/master/README_x86.md
• https://github.com/AastaNV/TRT_object_detection ("Update graphsurgeon converter")

[jkjung-avt]

I haven't tested the code with TensorRT 6.x before...

Are you running the code on a Jetson platform? Or a x86 PC?

[YoungSharp]

on x86 PC

[YoungSharp]

error happend when converting Input node.

[jkjung-avt]

I don't have an answer right now. My code was originally developed and tested on Jetson (arm64) platforms.

I might find time to test that later.

[YoungSharp]

i will keep trying, thanks.

[YoungSharp]

@jkjung-avt
Do you mind to share your training code with?
The code to train ssd_mobilenet_v2_egohands.pb
my code is latest "https://github.com/tensorflow/models/blob/master/research", and my model is at "https://drive.google.com/file/d/1e-9cb_sLmuevA4h1IPP029vUH3IxlIx1/view?usp=sharing"
my email address is [email protected]
Appreciated.

[jkjung-avt]

I already shared all my training code on GitHub. I was using a slightly older snapshot of TensorFlow Object Detection API. Please refer to README.md in the repository and the associated blog posts for details.

https://github.com/jkjung-avt/hand-detection-tutorial

[YoungSharp]

Thanks really

[jkjung-avt]

By the way, the problem you're facing might be due to some changes in TensorRT 6.0. I think the problem would likely go away if use TensorRT 5.x instead.

[YoungSharp]

Your model works well under Tensorrt 6.x;
anyway, i will give it a try.

[jkjung-avt]

I see, Thanks for the update.

[Ndron]

same problem

[Ndron]

@jkjung-avt Fixed , thanks

[aimuch]

@jkjung-avt Fixed , thanks

Hi， same problem，how did you solved it?
Thanks