classMask method

Float32List classMask(
  1. int classIndex
)

Returns a single-channel probability mask for the given class index.

Implementation

Float32List classMask(int classIndex) {
  if (classIndex < 0 || classIndex > 5) {
    throw RangeError.range(classIndex, 0, 5, 'classIndex');
  }
  final int numPixels = width * height;
  final result = Float32List(numPixels);
  for (int i = 0; i < numPixels; i++) {
    result[i] = internalClassData[i * 6 + classIndex];
  }
  return result;
}