Quantcast
Channel: TAO Toolkit - NVIDIA Developer Forums
Viewing all articles
Browse latest Browse all 497

How to build an ineractive UI on top of Nvidia TAO trained model that is in ".hdf5" format

$
0
0

My aim is to use the yolo_v4_object_detection model that is trained in TAO framework and saved in “.hdf5” format to do object detection.
It is giving good results on inferencing inside jupyter notebook.
How can I use one interactive UI to input one image of my choice and
the model will perform object detection and give us back the image with
bounding boxes , class name and confidence score.
Is there any dependencies on TAO framework ?
do I have to use Nvidia platform only, or , I can use any platform.
Basically how do I transfer my model from TAO framework and
integrate it into an interactive UI outside TAO ?
Can I use it directly in “.hdf5” format or do I have to convert it to any other format ?

In summary, I need guidance on converting the model to a compatible format and building an interactive UI to deploy it outside the TAO framework.

3 posts - 2 participants

Read full topic


Viewing all articles
Browse latest Browse all 497

Trending Articles