convertGrayscaleTensorBufferToImage static method

Image convertGrayscaleTensorBufferToImage(
  1. TensorBuffer buffer
)

Implementation

static Image convertGrayscaleTensorBufferToImage(TensorBuffer buffer) {
  // Convert buffer into Uint8 as needed.
  TensorBuffer uint8Buffer = buffer.getDataType() == TfLiteType.uint8
      ? buffer
      : TensorBuffer.createFrom(buffer, TfLiteType.uint8);

  final shape = uint8Buffer.getShape();
  final grayscale = ColorSpaceType.GRAYSCALE;
  grayscale.assertShape(shape);

  final image = Image.fromBytes(grayscale.getWidth(shape),
      grayscale.getHeight(shape), uint8Buffer.getBuffer().asUint8List(),
      format: Format.luminance);

  return image;
}