infrastructure

Documentation

Index

• Constants
• func InitDB(ctx context.Context, dns string) *gorm.DB
• type AI
  • func NewAI(invokeAIPath string) *AI
  • func (ai *AI) GenerateImage(ctx context.Context, seed uint, prompt string) ([]byte, error)
• type Config
  • func InitEnv() *Config
• type Webcam
  • func NewWebcam(deviceID string, resolution string) *Webcam
  • func (w *Webcam) Start(ctx context.Context, outputCh chan image.Image) error

Constants

const (
	TxtFilename  = "list.txt"
	ImageWidth   = 720
	ImageHeight  = 1080
	ImageSteps   = 50
	ImageUpscale = 4
)

стандартные параметры генерации, отлаженные на RTX3060 12Gb

const ReadTimeoutSeconds = 5

const V4L2_PIX_FMT_YUYV = 0x56595559 // https://github.com/blackjack/webcam/blob/master/examples/http_mjpeg_streamer/webcam.go

Functions

func InitDB

func InitDB(ctx context.Context, dns string) *gorm.DB

Types

type AI

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

AI - обёртка над Invoke.AI (который обёртка над Stable Diffusion), дающая создание картинок по prompt

func NewAI

func NewAI(invokeAIPath string) *AI

func (*AI) GenerateImage

func (ai *AI) GenerateImage(ctx context.Context, seed uint, prompt string) ([]byte, error)

type Config

type Config struct {
	// базовые параметры
	IsDebug      bool
	InvokeAIPath string
	DbDSN        string

	// hardware
	WebcamDeviceID        string
	WebcamFrameResolution string

	// службы
	CreatorActive bool
}

func InitEnv

func InitEnv() *Config

type Webcam

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

func NewWebcam

func NewWebcam(deviceID string, resolution string) *Webcam

func (*Webcam) Start

func (w *Webcam) Start(ctx context.Context, outputCh chan image.Image) error