Class NNModelExtensions
Inheritance
NNModelExtensions
Syntax
public static class NNModelExtensions
Methods
CreateWorker(NNModel, String[], String[], WorkerFactory.Device, Boolean)
Create a worker that will execute asset
using the best backend that is available for a given device
type.
This is just a convenience function that internally calls ModelLoader.Load
followed by `WorkerFactory.CreateWorker
.
asset
is the associated NNModel asset.
additionalOutputs
are the additional outputs to track but not directly specified by the model.
trimOutputs
are the outputs not discard even if they are specified by the model.
device
is the device type to run worker on. For example WorkerFactory.Device.GPU
specifies the fast GPU path.
verbose
will log scheduling of layers execution to the console (default == false).
Declaration
public static IWorker CreateWorker(this NNModel asset, string[] additionalOutputs, string[] trimOutputs, WorkerFactory.Device device = WorkerFactory.Device.Auto, bool verbose = false)
Parameters
Returns
CreateWorker(NNModel, WorkerFactory.Device, Boolean)
Create a worker that will execute asset
using the best backend that is available for a given device
type.
This is just a convenience function that internally calls ModelLoader.Load
followed by `WorkerFactory.CreateWorker
.
asset
is the associated NNModel asset.
device
is the preferred device for execution. For example WorkerFactory.Device.GPU
specifies the fast GPU path.
verbose
will log scheduling of layers execution to the console.
Declaration
public static IWorker CreateWorker(this NNModel asset, WorkerFactory.Device device = WorkerFactory.Device.Auto, bool verbose = false)
Parameters
Returns