module documentation
Utilities for working with SuperGradients.
Class |
|
FiftyOne wrapper around YOLO-NAS from https://github.com/Deci-AI/super-gradients. |
Class |
|
Configuration for running a TorchYoloNasModel . |
Function | convert |
Converts the given SuperGradients model into a FiftyOne model. |
Variable | logger |
Undocumented |
Variable | super |
Undocumented |
Function | _convert |
Undocumented |
Converts the given SuperGradients model into a FiftyOne model.
Parameters | |
model | a super_gradients.training.models.detection_models.yolo_nas model |
Returns | |
a fiftyone.core.models.Model | |
Raises | |
ValueError | if the model could not be converted |