fiftyone.utils.super_gradients.TorchYoloNasModelConfig

Modules Classes Names

class documentation

class TorchYoloNasModelConfig(fout.TorchImageModelConfig, fozm.HasZooModel): (source)

Constructor: TorchYoloNasModelConfig(d)

View In Hierarchy

Configuration for running a TorchYoloNasModel.

See fiftyone.utils.torch.TorchImageModelConfig for additional arguments.

Parameters
model	a preloaded `super_gradients.training.models.detection_models.yolo_nas` model to use
yolo_nas_model	the name of a YOLO-NAS model to use
pretrained	the pretrained version to use

Method	`__init__`	Undocumented
Instance Variable	`model`	Undocumented
Instance Variable	`pretrained`	Undocumented
Instance Variable	`yolo_nas_model`	Undocumented

Inherited from TorchImageModelConfig:

Instance Variable	`as_feature_extractor`	Undocumented
Instance Variable	`classes`	Undocumented
Instance Variable	`confidence_thresh`	Undocumented
Instance Variable	`cudnn_benchmark`	Undocumented
Instance Variable	`device`	Undocumented
Instance Variable	`embeddings_layer`	Undocumented
Instance Variable	`entrypoint_args`	Undocumented
Instance Variable	`entrypoint_fcn`	Undocumented
Instance Variable	`image_dim`	Undocumented
Instance Variable	`image_max_dim`	Undocumented
Instance Variable	`image_max_size`	Undocumented
Instance Variable	`image_mean`	Undocumented
Instance Variable	`image_min_dim`	Undocumented
Instance Variable	`image_min_size`	Undocumented
Instance Variable	`image_patch_size`	Undocumented
Instance Variable	`image_size`	Undocumented
Instance Variable	`image_std`	Undocumented
Instance Variable	`labels_path`	Undocumented
Instance Variable	`labels_string`	Undocumented
Instance Variable	`mask_targets`	Undocumented
Instance Variable	`mask_targets_path`	Undocumented
Instance Variable	`output_processor`	Undocumented
Instance Variable	`output_processor_args`	Undocumented
Instance Variable	`output_processor_cls`	Undocumented
Instance Variable	`ragged_batches`	Undocumented
Instance Variable	`raw_inputs`	Undocumented
Instance Variable	`skeleton`	Undocumented
Instance Variable	`transforms`	Undocumented
Instance Variable	`transforms_args`	Undocumented
Instance Variable	`transforms_fcn`	Undocumented
Instance Variable	`use_half_precision`	Undocumented

Inherited from Config (via TorchImageModelConfig):

Method __repr__ Undocumented

Inherited from HasZooModel (via TorchImageModelConfig, Config):

Method	`download_model_if_necessary`	Undocumented
Instance Variable	`model_path`	Undocumented
Class Method	`_get_model`	Undocumented

def __init__(self, d): (source) ¶

overrides fiftyone.utils.torch.TorchImageModelConfig.__init__

Undocumented

model: None = (source) ¶

overrides fiftyone.utils.torch.TorchImageModelConfig.model

Undocumented

pretrained: "coco" = (source) ¶

Undocumented

yolo_nas_model: "yolo_nas_l" = (source) ¶

Undocumented