Example application for using NVIDIA TAO DetectNet V2 models on the NXP i.MX 93 SoC.
-
model_conversion.ipynb: Jupyter Notebook to convert ONNX model to TensorFlow Lite model
-
create_nms_model.py
: Python script to createnms_model.tflite
-
Python Inference Application
main.py
: Inference Application entrydetect_net_v2_model.py
: Class to wrap DetectNet V2 TensorFlow Lite modelnms_model.py
: Class to wrap DetectNet V2 TensorFlow Lite model