fiftyone.utils.ultralytics.FiftyOneYOLOModelConfig

class documentation

class FiftyOneYOLOModelConfig(Config, fozm.HasZooModel): (source)

Known subclasses: fiftyone.utils.ultralytics.FiftyOneRTDETRModelConfig, fiftyone.utils.ultralytics.FiftyOneYOLOClassificationModelConfig, fiftyone.utils.ultralytics.FiftyOneYOLODetectionModelConfig, fiftyone.utils.ultralytics.FiftyOneYOLOOBBModelConfig, fiftyone.utils.ultralytics.FiftyOneYOLOPoseModelConfig, fiftyone.utils.ultralytics.FiftyOneYOLOSegmentationModelConfig

Constructor: FiftyOneYOLOModelConfig(d)

View In Hierarchy

Configuration for a FiftyOneYOLOModel.

Parameters
modelan ultralytics.YOLO model to use
model_namethe name of an ultralytics.YOLO model to load
model_paththe path to an ultralytics.YOLO model checkpoint
classesan optional list of classes
Method __init__ Undocumented
Instance Variable classes Undocumented
Instance Variable device Undocumented
Instance Variable model Undocumented
Instance Variable model_name Undocumented
Instance Variable model_path Undocumented

Inherited from Config:

Method __repr__ Undocumented

Inherited from HasZooModel (via Config):

Method download_model_if_necessary Undocumented
Class Method _get_model Undocumented