ViTObjectDetector constructor

ViTObjectDetector({
  1. required int imageSize,
  2. required int patchSize,
  3. required int numChannels,
  4. required int embedSize,
  5. required int numLayers,
  6. required int numHeads,
  7. required int numClasses,
  8. required int numQueries,
  9. required int embeddingDim,
})

Implementation

ViTObjectDetector({
  required int imageSize,
  required int patchSize,
  required int numChannels,
  required int embedSize,
  required int numLayers,
  required int numHeads,
  required int
      numClasses, // Number of object classes (identities + background)
  required this.numQueries,
  required this.embeddingDim, // New parameter
})  : backbone = ViTBackbone(
          imageSize: imageSize,
          patchSize: patchSize,
          numChannels: numChannels,
          embedSize: embedSize,
          numLayers: numLayers,
          numHeads: numHeads),
      detectionHead = ObjectDetectionHead(
          embedSize: embedSize,
          numClasses: numClasses,
          numQueries: numQueries,
          embeddingDim: embeddingDim);