Make sure the path to save onnx zip file exist (#2351)

Signed-off-by: LinGeLin <1057445597@qq.com>
onnx · Aug 22, 2024 · f85e88e · f85e88e
1 parent 0004163
commit f85e88e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tf2onnx/utils.py b/tf2onnx/utils.py
@@ -292,6 +292,9 @@ def save_onnx_model(save_path_root, onnx_file_name, feed_dict, model_proto, incl
 
 
 def save_onnx_zip(target_path, model_proto, external_tensor_storage):
+    dir_name = os.path.dirname(target_path)
+    if dir_name:
+        os.makedirs(dir_name, exist_ok=True)
     with zipfile.ZipFile(target_path, 'w') as z:
         z.writestr("__MODEL_PROTO.onnx", model_proto.SerializeToString())
         for k, v in external_tensor_storage.name_to_tensor_data.items():