PostProcessorObjectDetection constructor

PostProcessorObjectDetection(
  1. int numberOfClasses,
  2. int imageWidth,
  3. int imageHeight,
  4. ObjectDetectionModelType objectDetectionModelType,
)

Implementation

PostProcessorObjectDetection(
  this.numberOfClasses,
  this.imageWidth,
  this.imageHeight,
  this.objectDetectionModelType,
) {
  if (objectDetectionModelType == ObjectDetectionModelType.yolov5) {
    outputRow = 25200;
    outputColumn = (numberOfClasses + 5);
    modelOutputLength = outputRow * outputColumn;
  } else {
    outputRow = 8400;
    outputColumn = (numberOfClasses + 4);
    modelOutputLength = outputRow * outputColumn;
  }
}