neuralhelpers

Documentation

Index

• func CreateRandCropBatch(input, output []image.Image, h, w, resizedouth, resizedoutw int) (tin, tout *cvdh.Tensor4d)
• func Decoder(builder *gocunets.Builder, batchsize int32, outputchannel int32, ...) (mnet *gocunets.SimpleModuleNetwork)
• func Encoder(builder *gocunets.Builder, batchsize int32, outputchannel int32, ...) (mnet *gocunets.SimpleModuleNetwork)
• func GetImages(r []io.Reader) []image.Image
• type GPULoader
  • func CreateGpuLoader(dev cudart.Device, nbuffer int, SIBperbuff uint) (g *GPULoader, err error)
  • func (g *GPULoader) AsyncLoadMem(dest cutil.Pointer)
  • func (g *GPULoader) LoadBuffer(tensor *cvdh.Tensor4d)
  • func (g *GPULoader) SyncStream() error

Functions

func CreateRandCropBatch

func CreateRandCropBatch(input, output []image.Image, h, w, resizedouth, resizedoutw int) (tin, tout *cvdh.Tensor4d)

CreateRandCropBatch creates a RandCropBatch

func Decoder

func Decoder(
	builder *gocunets.Builder,
	batchsize int32,
	outputchannel int32,
	hiddenoutputchannels []int32,
	learningrates, decay1, decay2 float32,
	x, dx *gocunets.Tensor) (mnet *gocunets.SimpleModuleNetwork)

Decoder decodes a vector

func Encoder

func Encoder(
	builder *gocunets.Builder,
	batchsize int32,
	outputchannel int32,
	hiddenoutputchannels []int32,
	learningrates, decay1, decay2 float32,
	x *gocunets.Tensor) (mnet *gocunets.SimpleModuleNetwork)

Encoder encodes an image

func GetImages

func GetImages(r []io.Reader) []image.Image

GetImages gets images

Types

type GPULoader

type GPULoader struct {
	// contains filtered or unexported fields
}

GPULoader loads the gpu

func CreateGpuLoader

func CreateGpuLoader(dev cudart.Device, nbuffer int, SIBperbuff uint) (g *GPULoader, err error)

CreateGpuLoader creates a gpu loader

func (*GPULoader) AsyncLoadMem

func (g *GPULoader) AsyncLoadMem(dest cutil.Pointer)

func (*GPULoader) LoadBuffer

func (g *GPULoader) LoadBuffer(tensor *cvdh.Tensor4d)

func (*GPULoader) SyncStream

func (g *GPULoader) SyncStream() error