Versions in this module Expand all Collapse all v0 v0.9.2 Apr 21, 2024 v0.9.1 Apr 21, 2024 Changes in this version + func Byte2FloatImage(x *ts.Tensor) *ts.Tensor + func Float2ByteImage(x *ts.Tensor) *ts.Tensor + func HSV2RGB(x *ts.Tensor) *ts.Tensor + func RGB2HSV(x *ts.Tensor) *ts.Tensor + func RandomRotate(img *ts.Tensor, min, max float64) (*ts.Tensor, error) + func Rotate(img *ts.Tensor, angle float64) (*ts.Tensor, error) + type AffineOption func(*affineOptions) + func WithAffineDegree(degree []int64) AffineOption + func WithAffineFillValue(fillValue []float64) AffineOption + func WithAffineMode(mode string) AffineOption + func WithAffineScale(scale []float64) AffineOption + func WithAffineShear(shear []float64) AffineOption + func WithAffineTranslate(translate []float64) AffineOption + type Augment struct + func (a *Augment) Transform(image *ts.Tensor) *ts.Tensor + type CenterCrop struct + func (cc *CenterCrop) Forward(x *ts.Tensor) *ts.Tensor + type ColorJitter struct + func (c *ColorJitter) Forward(x *ts.Tensor) *ts.Tensor + type ColorOption func(*colorOptions) + func WithColorBrightness(v []float64) ColorOption + func WithColorContrast(v []float64) ColorOption + func WithColorHue(vals []float64) ColorOption + func WithColorSaturation(v []float64) ColorOption + type CutoutOption func(o *cutoutOptions) + func WithCutoutPvalue(p float64) CutoutOption + func WithCutoutRatio(ratio []float64) CutoutOption + func WithCutoutScale(scale []float64) CutoutOption + func WithCutoutValue(rgb []int64) CutoutOption + type DownSample struct + func (rs *DownSample) Forward(x *ts.Tensor) *ts.Tensor + type GaussianBlur struct + func (b *GaussianBlur) Forward(x *ts.Tensor) *ts.Tensor + type Grayscale struct + func NewGrayscale(outChanOpt ...int64) *Grayscale + func (gs *Grayscale) Forward(x *ts.Tensor) *ts.Tensor + type Normalize struct + func (n *Normalize) Forward(x *ts.Tensor) *ts.Tensor + type NormalizeOption func(*normalizeOptions) + func WithNormalizeMean(mean []float64) NormalizeOption + func WithNormalizeStd(std []float64) NormalizeOption + type Option func(o *Options) + func OneOf(pvalue float64, tfOpts ...Option) Option + func WithCenterCrop(size []int64) Option + func WithColorJitter(opts ...ColorOption) Option + func WithGaussianBlur(ks []int64, sig []float64) Option + func WithNormalize(opts ...NormalizeOption) Option + func WithRandRotate(minAngle, maxAngle float64) Option + func WithRandomAdjustSharpness(opts ...SharpnessOption) Option + func WithRandomAffine(opts ...AffineOption) Option + func WithRandomAutocontrast(p ...float64) Option + func WithRandomCrop(size []int64, padding []int64, paddingIfNeeded bool, paddingMode string) Option + func WithRandomCutout(opts ...CutoutOption) Option + func WithRandomEqualize(p ...float64) Option + func WithRandomGrayscale(pvalueOpt ...float64) Option + func WithRandomHFlip(pvalue float64) Option + func WithRandomInvert(pvalueOpt ...float64) Option + func WithRandomPerspective(opts ...PerspectiveOption) Option + func WithRandomPosterize(opts ...PosterizeOption) Option + func WithRandomSolarize(opts ...SolarizeOption) Option + func WithRandomVFlip(pvalue float64) Option + func WithResize(h, w int64) Option + func WithRotate(angle float64) Option + func WithZoomIn(v float64) Option + func WithZoomOut(v float64) Option + type Options struct + type PerspectiveOption func(*perspectiveOptions) + func WithPerspectiveMode(m string) PerspectiveOption + func WithPerspectivePvalue(p float64) PerspectiveOption + func WithPerspectiveScale(s float64) PerspectiveOption + func WithPerspectiveValue(v []float64) PerspectiveOption + type PosterizeOption func(*posterizeOptions) + func WithPosterizeBits(bits uint8) PosterizeOption + func WithPosterizePvalue(p float64) PosterizeOption + type RandRotateModule struct + func (rr *RandRotateModule) Forward(x *ts.Tensor) *ts.Tensor + type RandomAdjustSharpness struct + func (ras *RandomAdjustSharpness) Forward(x *ts.Tensor) *ts.Tensor + type RandomAffine struct + func (ra *RandomAffine) Forward(x *ts.Tensor) *ts.Tensor + type RandomAutocontrast struct + func (rac *RandomAutocontrast) Forward(x *ts.Tensor) *ts.Tensor + type RandomCrop struct + func (c *RandomCrop) Forward(x *ts.Tensor) *ts.Tensor + type RandomCutout struct + func (rc *RandomCutout) Forward(img *ts.Tensor) *ts.Tensor + type RandomEqualize struct + func (re *RandomEqualize) Forward(x *ts.Tensor) *ts.Tensor + type RandomGrayscale struct + func (rgs *RandomGrayscale) Forward(x *ts.Tensor) *ts.Tensor + type RandomHorizontalFlip struct + func (hf *RandomHorizontalFlip) Forward(x *ts.Tensor) *ts.Tensor + type RandomInvert struct + func (ri *RandomInvert) Forward(x *ts.Tensor) *ts.Tensor + type RandomPerspective struct + func (rp *RandomPerspective) Forward(x *ts.Tensor) *ts.Tensor + type RandomPosterize struct + func (rp *RandomPosterize) Forward(x *ts.Tensor) *ts.Tensor + type RandomResizedCrop struct + type RandomSolarize struct + func (rs *RandomSolarize) Forward(x *ts.Tensor) *ts.Tensor + type RandomVerticalFlip struct + func (vf *RandomVerticalFlip) Forward(x *ts.Tensor) *ts.Tensor + type ResizeModule struct + func (rs *ResizeModule) Forward(x *ts.Tensor) *ts.Tensor + type RotateModule struct + func (r *RotateModule) Forward(x *ts.Tensor) *ts.Tensor + type SharpnessOption func(*sharpnessOptions) + func WithSharpnessFactor(f float64) SharpnessOption + func WithSharpnessPvalue(p float64) SharpnessOption + type SolarizeOption func(*solarizeOptions) + func WithSolarizePvalue(p float64) SolarizeOption + func WithSolarizeThreshold(th float64) SolarizeOption + type Transformer interface + Transform func(x *ts.Tensor) *ts.Tensor + func Compose(opts ...Option) (Transformer, error) + type ZoomIn struct + func (rs *ZoomIn) Forward(x *ts.Tensor) *ts.Tensor + type ZoomOut struct + func (rs *ZoomOut) Forward(x *ts.Tensor) *ts.Tensor