From 0f31f9776a0b2b0fa663e40de5b8fd1d48f70f26 Mon Sep 17 00:00:00 2001 From: Legend Date: Sun, 23 Apr 2023 11:48:53 +0800 Subject: [PATCH] Update h5_to_tflite.py --- tools/h5_to_tflite.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/h5_to_tflite.py b/tools/h5_to_tflite.py index 39a30a6..0326046 100644 --- a/tools/h5_to_tflite.py +++ b/tools/h5_to_tflite.py @@ -34,7 +34,7 @@ def representative_data_gen(): continue valid_files.append(os.path.join(quant_dir, name)) if len(valid_files) == 0: - raise Exception("No valid files in quant_input dir {}, support format: ".format(quant_dir, valid_format)) + raise Exception("No valid files in quant_input dir {}, support format: {}".format(quant_dir, valid_format)) for path in valid_files: img = Image.open(path) img = np.array(img).astype(np.float32)