public Mnist(NNModel nnModel, BarracudaWorkerFactory.Type type)
{
bool verbose = false;
model = ModelLoader.Load(nnModel, verbose);
worker = BarracudaWorkerFactory.CreateWorker(type, model, verbose);
var kernels = ComputeShaderSingleton.Instance.kernels;
ops = new PrecompiledComputeOps(kernels, kernels[0]);
}
public TinyYolo(NNModel nnModel, BarracudaWorkerFactory.Type type)
{
bool verbose = false;
model = ModelLoader.Load(nnModel, verbose);
Debug.Log(model);
Debug.Log(model.inputs[0].name);
worker = BarracudaWorkerFactory.CreateWorker(type, model, verbose);
inputShape = new TensorShape(model.inputs[0].shape);
textureToTensor = new TextureToTensor();
resizeOptions = new TextureToTensor.ResizeOptions()
{
width = inputShape.width,
height = inputShape.height,
rotationDegree = 0,
flipX = false,
flipY = false,
aspectMode = TextureToTensor.AspectMode.Fill,
};
inputAspect = (float)inputShape.width / (float)inputShape.height;
results = new Result[25];
}
