public CoachWorker(CoachModel model)
{
this.Model = model;
this.Worker = WorkerFactory.CreateComputeWorker(model.Model);
this.Worker.PrepareForInput(new Dictionary <string, TensorShape>()
{
{ model.InputName, new TensorShape(1, model.ImageDims.InputSize, model.ImageDims.InputSize, 3) }
});
}
// Start is called before the first frame update
private void Start()
{
Debug.Log("Starting ONNX");
_runtimeModel = ModelLoader.Load(loadedModel);
_worker = WorkerFactory.CreateComputeWorker(_runtimeModel);
Debug.Log("ONNX Model Started");
foreach (var layer in _runtimeModel.layers)
{
Debug.Log(layer);
}
}
private void InitBarracuda()
{
m_RuntimeModel = ModelLoader.Load(_monoDepthONNX);
worker = WorkerFactory.CreateComputeWorker(m_RuntimeModel);
}
