ObjectDetectionHead constructor

ObjectDetectionHead({
  1. required int embedSize,
  2. required int numClasses,
})

Implementation

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