cls

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

type ColorFormatFunc func(r, g, b float32) (first, second, third float32)

type Config struct {
	ModelPath              string
	Width, Height          int
	OutputLabels           map[int][]string
	Shape                  Shape
	ColorFormatFunc        ColorFormatFunc
	ProcessImageFuncs      []ProcessImageFunc
	ProcessFloatImageFuncs []ProcessFloatImageFunc
}

type CoreMLConfig struct {
	Enabled bool
	//https://github.com/microsoft/onnxruntime/blob/291a5352b27ded5714e5748b381f2efb88f28fb9/include/onnxruntime/core/providers/coreml/coreml_provider_factory.h#L12
	Flags uint32
}

type CudaConfig struct {
	Enabled bool
	// https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html#configuration-options
	Options map[string]string
}

type DirectMLConfig struct {
	Enabled  bool
	DeviceID int
}

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

type OpenVINOConfig struct {
	Enabled bool
	//https://onnxruntime.ai/docs/execution-providers/OpenVINO-ExecutionProvider.html#summary-of-options
	Options map[string]string
}

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

type ProcessFloatImageFunc func(rgb float32) float32

type ProcessImageFunc func(img image.Image) image.Image

type SessionConfig struct {
	Cuda        CudaConfig
	TensortRT   TensorRTConfig
	CoreML      CoreMLConfig
	DirectML    DirectMLConfig
	CpuMemArena *bool
	MemPattern  *bool

	InterOpNumThreads int
	IntraOpNumThreads int
}

type Shape int

type TensorRTConfig struct {
	Enabled bool
	// https://onnxruntime.ai/docs/execution-providers/TensorRT-ExecutionProvider.html#execution-provider-options
	Options map[string]string
}