tensorimage

README

Image processing utilities

extracted from https://github.com/owulveryck/onnx-go/tree/master/internal/x/images

Documentation

Index

• func GrayToBCHW(img *image.Gray, dst tensor.Tensor) error
• func ImageToBCHW(img image.Image, dst tensor.Tensor) error
• func TensorToImg(t tensor.Tensor) (image.Image, error)

Functions

func GrayToBCHW

func GrayToBCHW(img *image.Gray, dst tensor.Tensor) error

GrayToBCHW convert an image to a BCHW tensor this function returns an error if:

• dst is not a pointer
• dst's shape is not 4
• dst' second dimension is not 1
• dst's third dimension != i.Bounds().Dy()
• dst's fourth dimension != i.Bounds().Dx()
• dst's type is not float32 or float64 (temporary)

func ImageToBCHW

func ImageToBCHW(img image.Image, dst tensor.Tensor) error

ImageToBCHW convert an image to a BCHW tensor this function returns an error if:

• dst is not a pointer
• dst's shape is not 4
• dst' second dimension is not 1
• dst's third dimension != i.Bounds().Dy()
• dst's fourth dimension != i.Bounds().Dx()
• dst's type is not float32 or float64 (temporary)

func TensorToImg

func TensorToImg(t tensor.Tensor) (image.Image, error)

TensorToImg turn a BCHW tensor into an image (BCHW with B=1)