Add support for .dat file as model input Change-Id: Idb70d178422acd9e25ae64ae52a80cf6f1ff9f71
diff --git a/build_tools/gen_mlmodel_input.py b/build_tools/gen_mlmodel_input.py index b6cdbbd..beabbad 100755 --- a/build_tools/gen_mlmodel_input.py +++ b/build_tools/gen_mlmodel_input.py
@@ -63,7 +63,8 @@ input_shape = [int(x) for x in args.input_shape.split(',')] input_ext = os.path.splitext(input_name)[1] is_audio = False - if (not input_ext) or (input_ext == '.bin') or (input_ext == '.raw'): + if (not input_ext) or (input_ext == '.bin') or \ + (input_ext == '.raw') or (input_ext == '.dat'): with open(input_name, mode='rb') as f: data = np.fromfile(f, dtype=np.uint8 if is_quant else np.float32) data = data[:np.prod(input_shape)].reshape(np.prod(input_shape))