Super Gradients Integration#
FiftyOne integrates natively with Deci AI’s SuperGradients library, so you can run inference with YOLO-NAS architectures on your FiftyOne datasets with just a few lines of code!
Setup#
To get started with
SuperGradients, just install
the super-gradients
package:
pip install super-gradients
Inference#
You can directly pass SuperGradients YOLO-NAS models to your FiftyOne dataset’s
apply_model()
method:
1import fiftyone as fo
2import fiftyone.zoo as foz
3
4from super_gradients.training import models
5
6dataset = foz.load_zoo_dataset("quickstart", max_samples=25)
7dataset.select_fields().keep_fields()
8
9model = models.get("yolo_nas_m", pretrained_weights="coco")
10# model = models.get("yolo_nas_l", pretrained_weights="coco")
11# model = models.get("yolo_nas_s", pretrained_weights="coco")
12
13dataset.apply_model(model, label_field="yolo_nas", confidence_thresh=0.7)
14
15session = fo.launch_app(dataset)
Model zoo#
SuperGradients YOLO-NAS is also available directly from the FiftyOne Model Zoo!
1import fiftyone as fo
2import fiftyone.zoo as foz
3
4model = foz.load_zoo_model("yolo-nas-torch")
5
6dataset = foz.load_zoo_dataset("quickstart")
7dataset.apply_model(model, label_field="yolo_nas")
8
9session = fo.launch_app(dataset)