ViTObjectDetector class

A complete Vision Transformer-based model for Face Detection and Recognition.

This model combines a ViT backbone for feature extraction with a detection head that predicts bounding boxes, class labels (identities + background), and a discriminative embedding for each detected face.

Inheritance

Constructors

ViTObjectDetector({required int imageSize, required int patchSize, required int numChannels, required int embedSize, required int numLayers, required int numHeads, required int numClasses, required int numQueries, required int embeddingDim})

Properties

backbone ViTBackbone
final
detectionHead ObjectDetectionHead
final
embeddingDim int
final
hashCode int
The hash code for this object.
no setterinherited
numQueries int
final
runtimeType Type
A representation of the runtime type of the object.
no setterinherited

Methods

forward(List<double> imageData) Map<String, List<ValueVector>>
Forward pass for the face detection and recognition model.
noSuchMethod(Invocation invocation) → dynamic
Invoked when a nonexistent method or property is accessed.
inherited
parameters() List<Value>
override
toString() String
A string representation of this object.
inherited
zeroGrad() → void
inherited

Operators

operator ==(Object other) bool
The equality operator.
inherited