inference

type Config struct {
	UserModelPath string
	LHost         string
}

type CreateRequest struct {
	// Image root path for training
	ImagePath string `json:"imagePath"`

	// Model meta information
	ModelPath   string `json:"modelPath"`
	ConfigFile  string `json:"configFile"`
	Description string `json:"desc"`

	Epochs int `json:"epochs"`

	Trial bool `json:"trial"`
}

type CreateResponse struct {
	ModelPath string `json:"modelPath" binding:"required"`
}

type InferLabel struct {
	Prob  float32 `json:"probability"`
	Label string  `json:"label"`
}

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