astiocr

type ConfigurationColor struct {
	Background astiimage.RGBA   `toml:"background"`
	Fonts      []astiimage.RGBA `toml:"fonts"`
}

type ConfigurationDetector struct {
	// Path to the model
	ModelPath string `toml:"model_path"`
}

type ConfigurationFont struct {
	File          string  `toml:"file"`
	PositionRatio float64 `toml:"position_ratio"`
}

type ConfigurationImage struct {
	Height int `toml:"height"`
	Width  int `toml:"width"`
}

type ConfigurationTrainer struct {
	// Path to the cache directory
	CacheDirectoryPath string `toml:"cache_directory_path"`

	// Number of images generated for both training and test purposes
	Count int `toml:"count"`

	// Color options
	Colors []ConfigurationColor `toml:"colors"`

	// Font options
	Fonts []ConfigurationFont `toml:"fonts"`

	// Image options
	Image ConfigurationImage `toml:"image"`

	// Path to the output directory
	OutputDirectoryPath string `toml:"output_directory_path"`

	// Path to the python binary
	PythonBinaryPath string `toml:"python_binary_path"`

	// Path to the scripts directory
	ScriptsDirectoryPath string `toml:"scripts_directory_path"`

	// Show box around labels
	ShowBox bool `toml:"show_box"`

	// Show entire grid
	ShowGrid bool `toml:"show_grid"`

	// Path to the tensorflow models directory
	TensorFlowModelsDirectoryPath string `toml:"tensorflow_models_directory_path"`

	// The proportion of test data in the generated images
	TestDataProportion float64 `toml:"test_data_proportion"`
}

type DetectionBox struct {
	X1, X2 float64
	Y1, Y2 float64
}

type DetectionResult struct {
	Box         DetectionBox
	Label       string
	Probability float64
}

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

type GatherSummary struct {
	Images []GatherSummaryImage `json:"images"`
}

type GatherSummaryBox struct {
	Label      string `json:"label"`
	LabelIndex int    `json:"label_index"`
	X0         int    `json:"x0"`
	X1         int    `json:"x1"`
	Y0         int    `json:"y0"`
	Y1         int    `json:"y1"`
}

type GatherSummaryImage struct {
	Height int                `json:"height"`
	Boxes  []GatherSummaryBox `json:"boxes"`
	Path   string             `json:"path"`
	Width  int                `json:"width"`
}

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