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,
})

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 (excluding background)
  required this.numQueries, // New parameter
})  : backbone = ViTBackbone(
          imageSize: imageSize,
          patchSize: patchSize,
          numChannels: numChannels,
          embedSize: embedSize,
          numLayers: numLayers,
          numHeads: numHeads),
      detectionHead = ObjectDetectionHead(
          embedSize: embedSize,
          numClasses: numClasses,
          numQueries: numQueries);