在本教程中,我们将演示如何使用该model.predict()
方法进行对象检测任务。
本教程使用的模型是YOLO-NAS ,在COCO 数据集上进行预训练,其中包含 80 个对象类别。
警告:如果您在不继承任何 SuperGradients 数据集的数据集上训练模型,则在运行模型之前需要执行一些额外的步骤。
请注意,该 model.predict()
方法目前仅适用于检测任务。
在本教程中,我们将演示如何使用该model.predict()
方法进行对象检测任务。
本教程使用的模型是YOLO-NAS ,在COCO 数据集上进行预训练,其中包含 80 个对象类别。
警告:如果您在不继承任何 SuperGradients 数据集的数据集上训练模型,则在运行模型之前需要执行一些额外的步骤。
请注意,该 model.predict()
方法目前仅适用于检测任务。