How to Use My Trained Yolo Model's Weights With TAO Toolkit?

Accelerated Computing Intelligent Video Analytics TAO Toolkit
1

Hi. I have a trained Yolov4 model with .weights and .cfg files. Can i or should i use this model with TAO?

Thanks.

2

Currently, it is not supported.
BTW, next TAO release can support PyTorch model of Unet or Classification network.
Other object detection models will be included in the future.

1 Like
3

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