ObjectDetectionHead constructor

ObjectDetectionHead({ required int embedSize, required int numClasses, required int numQueries, })

Implementation

ObjectDetectionHead({
  required this.embedSize,
  required this.numClasses,
  required this.numQueries,
})  : bboxRegressionHead =
          Layer.fromNeurons(embedSize, numQueries * numBoxCoords),
      classPredictionHead = Layer.fromNeurons(embedSize,
          numQueries * (numClasses + 1));