imageloader

Documentation

Index

• Constants
• func BuildLabelVocabularyFromTgz(fn string) (map[string]int, error)
• type ImageLoader
  • func New(fn string, vocab map[string]int, trans *transforms.ComposeTransformer, ...) (*ImageLoader, error)
  • func (p *ImageLoader) Err() error
  • func (p *ImageLoader) Minibatch() (torch.Tensor, torch.Tensor)
  • func (p *ImageLoader) Scan() bool

Constants

const GRAY string = "gray"

GRAY color

const RGB string = "rgb"

RGB color

Functions

func BuildLabelVocabularyFromTgz

func BuildLabelVocabularyFromTgz(fn string) (map[string]int, error)

BuildLabelVocabularyFromTgz build a label vocabulary from the image tgz file

Types

type ImageLoader

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

ImageLoader struct

func New

func New(fn string, vocab map[string]int, trans *transforms.ComposeTransformer,
	mbSize, bufSize int, seed int64, pinMemory bool, colorSpace string) (*ImageLoader, error)

New returns an ImageLoader

func (*ImageLoader) Err

func (p *ImageLoader) Err() error

Err returns the error during the scan process, if there is any. io.EOF is not considered an error.

func (*ImageLoader) Minibatch

func (p *ImageLoader) Minibatch() (torch.Tensor, torch.Tensor)

Minibatch returns a minibatch with data and label Tensor

func (*ImageLoader) Scan

func (p *ImageLoader) Scan() bool

Scan return false if no more data