nsfw

package module

v0.0.0-...-d5b32f8 Latest Latest Go to latest Published: Aug 3, 2022 License: MIT Imports: 16 Imported by: 0

Details

A Go port of the GantMan's nsfw_model for nudity detection and classification using TensorFlow

View Source

const (
	DefaultCachePath = "./.models/"
)

View Source

const (
	ImageDimensions = 224
)

View Source

var (
	ErrNoneCached = errors.New("no cached models")
)

This section is empty.

type Path string

func GetLatestModelPath() (Path, error)

func GetLocalModelPath() (Path, error)

func (p Path) GetModel() *tg.Model

func (p Path) String() string

type Prediction struct {
	Drawings float32
	Hentai   float32
	Neutral  float32
	Porn     float32
	Sexy     float32
}

func (p Prediction) Describe() string

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

func NewLatestPredictor() (*Predictor, error)

func NewPredictor(model *tg.Model) *Predictor

func (p *Predictor) NewImage(filepath string, channels int64) *image.Image

func (p *Predictor) Predict(img *image.Image) Prediction

func (p *Predictor) UseScope(s *op.Scope)

Path	Synopsis
example