YOLO-NAS detailed tutorial--how to use the model.predict() method for object detection tasks

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.

Detect objects in multiple images

Load the model and prepare the image

Guess you like

Origin blog.csdn.net/tianqiquan/article/details/132890334