model.go

// All material is licensed under the Apache License Version 2.0, January 2004
// http://www.apache.org/licenses/LICENSE-2.0

package dlinfer

import "os"

// #cgo CXXFLAGS: -std=c++11 -I/usr/include -I/opt/intel/deep_learning_sdk_2016.1.0.861/deployment_tools/inference_engine/include -I/opt/intel/deep_learning_sdk_2016.1.0.861/deployment_tools/inference_engine/samples/format_reader
// #cgo LDFLAGS: -L/opt/intel/deep_learning_sdk_2016.1.0.861/deployment_tools/inference_engine/bin/intel64/lib -L/opt/intel/deep_learning_sdk_2016.1.0.861/deployment_tools/inference_engine/lib/intel64 -ldl -linference_engine -lformat_reader
import "C"

// Configurator inncludes the necessary pieces of an
// Inferenence Engine Configurator, as used in the Intel
// Deep Learning iSDK.
type Configurator struct {
	modelFile  string
	pluginPath string
	pluginName string
	labelFile  string
}

// NewConfigurator creates a new configurator for a particular
// trained model.
func NewConfigurator(modelFile, pluginPath, pluginName, labelFile string) (*Configurator, error) {

	// Validate the model file.
	if _, err := os.Stat(modelFile); os.IsNotExist(err) {
		return nil, err
	}

	// Return the configurator.
	return &Configurator{
		modelFile:  modelFile,
		pluginPath: pluginPath,
		pluginName: pluginName,
		labelFile:  labelFile,
	}, nil
}

// LoadImage loads an image as input to an inference.
func LoadImage() {

}