ObjectDetectionHead constructor

ObjectDetectionHead({
  1. required int embedSize,
  2. required int numClasses,
  3. required int numQueries,
  4. required int embeddingDim,
})

Implementation

ObjectDetectionHead({
  required this.embedSize,
  required this.numClasses,
  required this.numQueries,
  required this.embeddingDim, // New parameter for embedding dimension
})  : bboxRegressionHead =
          Layer.fromNeurons(embedSize, numQueries * numBoxCoords),
      classPredictionHead = Layer.fromNeurons(embedSize,
          numQueries * (numClasses + 1)), // +1 for background class
      faceEmbeddingHead = Layer.fromNeurons(embedSize,
          numQueries * embeddingDim);