steps.ultralytics.model.loader

loader

Functions:

Name	Description
load_ultralytics_model	Load an Ultralytics YOLO model with the specified weights and configurations.

load_ultralytics_model(pretrained_weights_name, trained_weights_name=None, config_name=None, exported_weights_name=None)

Load an Ultralytics YOLO model with the specified weights and configurations.

This step: - Retrieves the current training context from the pipeline. - Initializes an UltralyticsModel using metadata from the experiment and optional weight/config names. - Downloads the pretrained weights and loads them using Ultralytics' YOLO API. - Sets the loaded model into the UltralyticsModel instance for downstream training or inference.

Parameters:

Name	Type	Description	Default
pretrained_weights_name	str	The name of the pretrained weights file to load.	required
trained_weights_name	str	Name of the trained weights (if resuming training or fine-tuning).	None
config_name	str	Name of the configuration file used for the model.	None
exported_weights_name	str	Name of the exported weights (e.g., for inference/export steps).	None

Returns:

Name	Type	Description
UltralyticsModel	UltralyticsModel	An initialized model with its architecture and weights loaded.