bundle

Documentation

Index

• func GetRuntime() (*ort.Runtime, error)
• type Detector
  • func New(opts *ort.SessionOptions) (*Detector, error)
  • func (d *Detector) Close() error
  • func (d *Detector) Detect(ctx context.Context, img image.Image) (nsfw.Labels, error)

Functions

func GetRuntime

func GetRuntime() (*ort.Runtime, error)

Types

type Detector

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

func New

func New(opts *ort.SessionOptions) (*Detector, error)

func (*Detector) Close

func (d *Detector) Close() error

func (*Detector) Detect

func (d *Detector) Detect(ctx context.Context, img image.Image) (nsfw.Labels, error)