How to run tensorflow mobile net V2 model on Jetson Nano

Robotics & Edge Computing Jetson Systems Jetson Nano
3

Hi,

Please check this GitHub for information:

You may need to update the config.py into this version since TensorFlow has some update:

Then, the class number information can be customized as the following:

# config.py
...
NMS = gs.create_plugin_node(name="NMS", op="NMS_TRT",
    ...
    numClasses=3,
    ...

Thanks.