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

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)
})  : backbone = ViTBackbone(
          imageSize: imageSize,
          patchSize: patchSize,
          numChannels: numChannels,
          embedSize: embedSize,
          numLayers: numLayers,
          numHeads: numHeads),
      detectionHead = ObjectDetectionHead(
          embedSize: embedSize, numClasses: numClasses);