How to configure TensorFlow 2.0 launcher¶
TensorFlow 2.0 launcher is one of the supported wrappers for easily launching models within Accuracy Checker tool. This launcher allows to execute models in saved models format.
Important note : this launcher supports only TensorFlow 2.0 and higher version and works with saved models format, which contains tf.function object. For other model saving formats please use TensorFlow launcher
For enabling TensorFlow 2.0 launcher you need to add
framework: tf2 in launchers section of your configuration file and provide following parameters:
saved_model_dir- path to directory with saved model.
adapter- approach how raw output will be converted to representation of dataset problem, some adapters can be specific to framework. You can find detailed instruction how to use adapters here.
device- specifies which device will be used for infer (
Specifying model inputs in config.¶
In case when you model has several inputs you should provide list of input layers in launcher config section using key
inputs. Each input description should has following info:
name- input layer name in network
type- type of input values, it has impact on filling policy. Available options:
CONST_INPUT- input will be filled using constant provided in config. It also requires to provide
IMAGE_INFO- specific key for setting information about input shape to layer (used in Faster RCNN based topologies). You do not need to provide
value, because it will be calculated in runtime. Format value is list with
Nelements of the form
[H, W, S], where
Nis batch size,
H- original image height,
W- original image width,
S- scale of original image (default 1).
ORIG_IMAGE_INFO- specific key for setting information about original image size before preprocessing.
IGNORE_INPUT- input which should be stayed empty during evaluation.
INPUT- network input for main data stream (e. g. images). If you have several data inputs, you should provide regular expression for identifier as
valuefor specifying which one data should be provided in specific input.
Optionally you can determine
shapeof input and
layoutin case when your model was trained with non-standard data layout (For TensorFlow default layout is
FP16- signed shot,
U8- unsigned char,
U16- unsigned short int,
I8- signed char,
I16- short int,
I64- long int).
TensorFlow 2 launcher config example:
launchers: - framework: tf2 device: CPU saved_model_dir: saved_model/ adapter: classification