In this tutorial, we will demonstrate how to use this model.predict()
method for object detection tasks.
The model used in this tutorial is YOLO-NAS, pre-trained on the COCO dataset , which contains 80 object categories.
Warning : If you train a model on a dataset that does not inherit any SuperGradients datasets, you need to perform some additional steps before running the model.
Note that this model.predict()
method currently only works for detection tasks.