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 AlexNet(p *nn.Path, nclasses int64) ts.ModuleT + func Augmentation(t *ts.Tensor, flip bool, crop int64, cutout int64) *ts.Tensor + func DenseNet121(p *nn.Path, nclasses int64) ts.ModuleT + func DenseNet161(p *nn.Path, nclasses int64) ts.ModuleT + func DenseNet169(p *nn.Path, nclasses int64) ts.ModuleT + func DenseNet201(p *nn.Path, nclasses int64) ts.ModuleT + func EfficientNetB0(p *nn.Path, nclasses int64) ts.ModuleT + func EfficientNetB1(p *nn.Path, nclasses int64) ts.ModuleT + func EfficientNetB2(p *nn.Path, nclasses int64) ts.ModuleT + func EfficientNetB3(p *nn.Path, nclasses int64) ts.ModuleT + func EfficientNetB4(p *nn.Path, nclasses int64) ts.ModuleT + func EfficientNetB5(p *nn.Path, nclasses int64) ts.ModuleT + func EfficientNetB6(p *nn.Path, nclasses int64) ts.ModuleT + func EfficientNetB7(p *nn.Path, nclasses int64) ts.ModuleT + func InceptionV3(p *nn.Path, nclasses int64) ts.ModuleT + func Load(path string) (*ts.Tensor, error) + func LoadAndResize(path string, outW int64, outH int64) (*ts.Tensor, error) + func LoadDir(dir string, outW int64, outH int64) (*ts.Tensor, error) + func MobileNetV2(p *nn.Path, nclasses int64) ts.ModuleT + func RandomCrop(t *ts.Tensor, pad int64) *ts.Tensor + func RandomCutout(t *ts.Tensor, sz int64) *ts.Tensor + func RandomFlip(t *ts.Tensor) *ts.Tensor + func ResNet101(path *nn.Path, numClasses int64) ts.ModuleT + func ResNet101NoFinalLayer(path *nn.Path) ts.ModuleT + func ResNet150NoFinalLayer(path *nn.Path) ts.ModuleT + func ResNet152(path *nn.Path, numClasses int64) ts.ModuleT + func ResNet18(path *nn.Path, numClasses int64) nn.FuncT + func ResNet18NoFinalLayer(path *nn.Path) nn.FuncT + func ResNet34(path *nn.Path, numClasses int64) nn.FuncT + func ResNet34NoFinalLayer(path *nn.Path) nn.FuncT + func ResNet50(path *nn.Path, numClasses int64) ts.ModuleT + func ResNet50NoFinalLayer(path *nn.Path) ts.ModuleT + func Resize(t *ts.Tensor, outW int64, outH int64) (*ts.Tensor, error) + func ResizePreserveAspectRatio(t *ts.Tensor, outW int64, outH int64) (*ts.Tensor, error) + func Save(tensor *ts.Tensor, path string) error + func SqueezeNetV1_0(p *nn.Path, nclasses int64) ts.ModuleT + func SqueezeNetV1_1(p *nn.Path, nclasses int64) ts.ModuleT + func VGG11(path *nn.Path, nclasses int64) *nn.SequentialT + func VGG11BN(path *nn.Path, nclasses int64) *nn.SequentialT + func VGG13(path *nn.Path, nclasses int64) *nn.SequentialT + func VGG13BN(path *nn.Path, nclasses int64) *nn.SequentialT + func VGG16(path *nn.Path, nclasses int64) *nn.SequentialT + func VGG16BN(path *nn.Path, nclasses int64) *nn.SequentialT + func VGG19(path *nn.Path, nclasses int64) *nn.SequentialT + func VGG19BN(path *nn.Path, nclasses int64) *nn.SequentialT + type BlockArgs struct + ExpandRatio int64 + InputFilters int64 + KernelSize int64 + NumRepeat int64 + OutputFilter int64 + SeRatio float64 + Stride int64 + type Dataset struct + Labels int64 + TestImages *ts.Tensor + TestLabels *ts.Tensor + TrainImages *ts.Tensor + TrainLabels *ts.Tensor + func CFLoadDir(dir string) *Dataset + func LoadMNISTDir(dir string) *Dataset + func (ds *Dataset) TestIter(batchSize int64) *ts.Iter2 + func (ds *Dataset) TrainIter(batchSize int64) *ts.Iter2 + type ImageNet struct + func NewImageNet() *ImageNet + func (in *ImageNet) ClassCount() int64 + func (in *ImageNet) Classes() []string + func (in *ImageNet) LoadFromDir(path string) (*Dataset, error) + func (in *ImageNet) LoadImage(path string) (*ts.Tensor, error) + func (in *ImageNet) LoadImageAndResize(path string, w, h int64) (*ts.Tensor, error) + func (in *ImageNet) LoadImageAndResize224(path string) (*ts.Tensor, error) + func (in *ImageNet) Normalize(tensor *ts.Tensor) (*ts.Tensor, error) + func (in *ImageNet) SaveImage(tensor *ts.Tensor, path string) error + func (in *ImageNet) Top(input *ts.Tensor, k int64) []TopItem + func (in *ImageNet) UnNormalize(tensor *ts.Tensor) (*ts.Tensor, error) + type TopItem struct + Label string + Pvalue float64
Go to main page