YOLO-NAS详细教程--如何使用该model.predict()方法进行对象检测任务

在本教程中,我们将演示如何使用该model.predict()方法进行对象检测任务。

本教程使用的模型是YOLO-NAS ,在COCO 数据集上进行预训练,其中包含 80 个对象类别。

警告:如果您在不继承任何 SuperGradients 数据集的数据集上训练模型,则在运行模型之前需要执行一些额外的步骤。

请注意,该 model.predict()方法目前仅适用于检测任务。

检测多个图像中的对象

加载模型并准备图像

猜你喜欢

转载自blog.csdn.net/tianqiquan/article/details/132890334
今日推荐