diff --git a/build_tools/gen_mlmodel_input.py b/build_tools/gen_mlmodel_input.py
index bbcd8d0..e40c8f4 100755
--- a/build_tools/gen_mlmodel_input.py
+++ b/build_tools/gen_mlmodel_input.py
@@ -59,7 +59,7 @@
         raise ValueError("Input shape < 3 dimensions")
     input_ext = os.path.splitext(input_name)[1]
     is_audio = False
-    if (not input_ext) or (input_ext == '.bin'):
+    if (not input_ext) or (input_ext == '.bin') or (input_ext == '.raw'):
         with open(input_name, mode='rb') as f:
             input = np.fromfile(f, dtype=np.uint8 if is_quant else np.float32)
             input = input[:np.prod(input_shape)].reshape(np.prod(input_shape))