Class ImageEncoder<T>

Namespace

AiDotNet.Diffusion.Models

Assembly

AiDotNet.dll

Image encoder for extracting features from reference images.

public class ImageEncoder<T>

Type Parameters

T

The numeric type.

Inheritance

object

ImageEncoder<T>

Inherited Members

object.Equals(object)

object.Equals(object, object)

object.GetHashCode()

object.GetType()

object.MemberwiseClone()

object.ReferenceEquals(object, object)

object.ToString()

Constructors

ImageEncoder(int, int, int, int, int, int?)

Initializes a new ImageEncoder.

public ImageEncoder(int imageSize = 224, int patchSize = 16, int embedDim = 768, int numLayers = 12, int numHeads = 12, int? seed = null)

Parameters

imageSize int

patchSize int

embedDim int

numLayers int

numHeads int

seed int?

Properties

ParameterCount

Gets the number of parameters.

public int ParameterCount { get; }

Property Value

int

Methods

Encode(Tensor<T>)

Encodes an image into feature embeddings.

public Tensor<T> Encode(Tensor<T> image)

Parameters

image Tensor<T>

Returns

Tensor<T>

GetParameters()

Gets all parameters.

public Vector<T> GetParameters()

Returns

Vector<T>

SetParameters(Vector<T>)

Sets all parameters.

public void SetParameters(Vector<T> parameters)

Parameters

parameters Vector<T>