convertGrayscaleTensorBufferToImage static method
Image
convertGrayscaleTensorBufferToImage(
- 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;
}