mnist

README

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

Documentation

Index

• Constants
• func ToGrayScale(m [][]uint8) image.Image
• type Files
• type ImageSet
  • func (img *ImageSet) GetFlatImage(offset int) []uint8
  • func (img *ImageSet) GetImage(offset int) [][]uint8
• type LabelSet
  • func (l *LabelSet) GetLabel(offset int) uint8
• type Reader
  • func New(locs *Files) (*Reader, error)

Constants

const (
	ImageMagic int32 = 2051
	LabelMagic int32 = 2049
)

Functions

func ToGrayScale

func ToGrayScale(m [][]uint8) image.Image

Types

type Files

type Files struct {
	TrainingImagesLoc string
	TrainingLabelsLoc string
	TestingImagesLoc  string
	TestingLabelsLoc  string
}

type ImageSet

type ImageSet struct {
	Count      int
	Rows, Cols int
	Images     []uint8
}

func (*ImageSet) GetFlatImage

func (img *ImageSet) GetFlatImage(offset int) []uint8

func (*ImageSet) GetImage

func (img *ImageSet) GetImage(offset int) [][]uint8

type LabelSet

type LabelSet struct {
	Count  int
	Labels []uint8
}

func (*LabelSet) GetLabel

func (l *LabelSet) GetLabel(offset int) uint8

type Reader

type Reader struct {
	TrainingImages, TestingImages *ImageSet
	TrainingLabels, TestingLabels *LabelSet
}

func New

func New(locs *Files) (*Reader, error)