dogsvscats

Nov 30, 2023

Changes in this version

+ const PreGeneratedTrainPairFileName

type Configuration

+ NumSamples int

+ YieldImagePairs bool

type Dataset

+ func (ds *Dataset) WithImagePairs(yieldPairs bool) *Dataset

type PreGeneratedDataset

+ func (pds *PreGeneratedDataset) WithImagePairs(pairFilePath string) *PreGeneratedDataset

Jun 5, 2023

Changes in this version

type Configuration

+ EvalBatchSize int

type PreGeneratedDataset

+ func (pds *PreGeneratedDataset) WithMaxSteps(numSteps int) *PreGeneratedDataset

Apr 23, 2023

Changes in this version

+ const DownloadChecksum

+ const DownloadURL

+ const InvalidSubDir

+ const LocalZipDir

+ const LocalZipFile

+ const PreGeneratedTrainEvalFileName

+ const PreGeneratedTrainFileName

+ const PreGeneratedValidationFileName

+ var BadCatImages = map[int]bool

+ var BadDogImages = map[int]bool

+ var BadImages = [2]map[int]bool

+ var DefaultConfig = &Configuration

+ var ImgSubDirs = []string

+ var MaxCount = 12500

+ var NumCats = MaxCount - len(BadCatImages)

+ var NumDogs = MaxCount - len(BadDogImages)

+ var NumValid = [2]int

+ func AssertNoError(err error)

+ func BytesToTensor(buffer []byte, numImages, width, height int) (t *tensor.Local)

+ func CreateDatasets(config *Configuration) (trainDS, trainEvalDS, validationEvalDS train.Dataset)

+ func Download(baseDir string) error

+ func FilterValidImages(baseDir string) error

+ func GetImageFromFilePath(imagePath string) (image.Image, error)

+ func ImagesToTensor(imgs []image.Image) (t *tensor.Local)

+ func PreGenerate(config *Configuration, numEpochsForTraining int)

+ func PrefilterValidImages(baseDir string) error

+ func ResizeWithPadding(img image.Image, width, height int) image.Image

+ func TensorToGoImage(config *Configuration, images tensor.Tensor, exampleNum int) *image.NRGBA

+ type Configuration struct

+ AngleStdDev float64

+ BatchSize int

+ BufferSize int

+ DType shapes.DType

+ DataDir string

+ FlipRandomly bool

+ FoldsSeed int32

+ ForceOriginal bool

+ ModelImageSize int

+ NumFolds int

+ TrainFolds []int

+ UseParallelism bool

+ ValidationFolds []int

+ type Dataset struct

+ BaseDir string

+ var AssertDatasetIsTrainDataset *Dataset

+ func NewDataset(name, baseDir string, batchSize int, infinite bool, shuffle *rand.Rand, ...) *Dataset

+ func (ds *Dataset) Augment(img image.Image) image.Image

+ func (ds *Dataset) Name() string

+ func (ds *Dataset) Reset()

+ func (ds *Dataset) Save(writer io.Writer, numEpochs int, verbose bool) error

+ func (ds *Dataset) Yield() (spec any, inputs, labels []tensor.Tensor, err error)

+ func (ds *Dataset) YieldImages() (images []image.Image, labels []DorOrCat, err error)

+ type DorOrCat int8

+ const Cat

+ const Dog

+ func (t DorOrCat) String() string

+ type PreGeneratedDataset struct

+ func NewPreGeneratedDataset(name, filePath string, batchSize int, infinite bool, width, height int, ...) *PreGeneratedDataset

+ func (pds *PreGeneratedDataset) Name() string

+ func (pds *PreGeneratedDataset) Reset()

+ func (pds *PreGeneratedDataset) Yield() (spec any, inputs, labels []tensor.Tensor, err error)