nsfw

README

NSFW Image Classifier

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

Documentation

Index

• Constants
• Variables
• type Path
  • func GetLatestModelPath() (Path, error)
  • func GetLocalModelPath() (Path, error)
  • func (p Path) GetModel() *tg.Model
  • func (p Path) String() string
• type Prediction
  • func (p Prediction) Describe() string
• type Predictor
  • 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)

Constants

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

const (
	ImageDimensions = 224
)

Variables

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

Types

type Path

type Path string

func GetLatestModelPath

func GetLatestModelPath() (Path, error)

func GetLocalModelPath

func GetLocalModelPath() (Path, error)

func (Path) GetModel

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

func (Path) String

func (p Path) String() string

type Prediction

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

func (Prediction) Describe

func (p Prediction) Describe() string

type Predictor

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

func NewLatestPredictor

func NewLatestPredictor() (*Predictor, error)

func NewPredictor

func NewPredictor(model *tg.Model) *Predictor

func (*Predictor) NewImage

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

func (*Predictor) Predict

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

func (*Predictor) UseScope

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