Class PretrainedRegistry

Namespace

AiDotNet.ComputerVision.Weights

Assembly

AiDotNet.dll

Registry of pre-trained model weights with their download URLs.

public static class PretrainedRegistry

Inheritance

object

PretrainedRegistry

Inherited Members

object.Equals(object)

object.Equals(object, object)

object.GetHashCode()

object.GetType()

object.MemberwiseClone()

object.ReferenceEquals(object, object)

object.ToString()

Methods

GetAvailableModels()

Gets all available pretrained model keys.

public static IEnumerable<string> GetAvailableModels()

Returns

IEnumerable<string>

Collection of model keys.

GetDetectionModelKey(DetectionArchitecture, ModelSize, string)

Generates the standard model key for detection models.

public static string GetDetectionModelKey(DetectionArchitecture architecture, ModelSize size, string dataset = "coco")

Parameters

architecture DetectionArchitecture

The detection architecture.

size ModelSize

The model size.

dataset string

The training dataset (default: "coco").

Returns

string

The model key.

GetUrl(string)

Gets the URL for a pretrained model.

public static string? GetUrl(string modelKey)

Parameters

modelKey string

The model key (e.g., "yolov8m-coco").

Returns

string

The download URL, or null if not found.

Register(string, string)

Registers a custom weight URL.

public static void Register(string modelKey, string url)

Parameters

modelKey string

The model key.

url string

The download URL.