Can I use my own models in TAO, like a trained yolov5 or yolov8?

Accelerated Computing Intelligent Video Analytics TAO Toolkit
1

I have asked similar problem before, how about this problem now?

(Can I use tao toolkit on my own models)

3

TAO is already open-sourced since 5.0 version. See the bottom of NVIDIA Corporation · GitHub.
For YOLOv4, please refer to GitHub - NVIDIA/tao_tensorflow1_backend: TAO Toolkit deep learning networks with TensorFlow 1.x backend branch.
For deployment, please refer to GitHub - NVIDIA/tao_deploy: Package for deploying deep learning models from TAO Toolkit.
YOLOv5 and YOLOv8 are not available in TAO. You can leverage the TAO open source code to deploy.

1 Like
4

There is no update from you for a period, assuming this is not an issue anymore. Hence we are closing this topic. If need further support, please open a new one. Thanks

5

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.