/mnist

MNIST reader for Go

Primary LanguageGoMIT LicenseMIT

Go MNIST

Installing

go get github.com/sno6/mnist

Getting started

// Read images & labels from local ubyte files.
r, err := mnist.New(&mnist.Files{
    TrainingImagesLoc: imagesLoc,
    TrainingLabelsLoc: labelsLoc,
})

...

// Use the reader to access images & labels.
img := r.TrainingImages.GetImage(1)
label := r.TrainingLabels.GetLabel(1)

...

Example

There's also a working code example here