fix model export. Update tf2onnx to 1.9.2

2024-03-22 13:10:55 +08:00 · 2021-11-20 16:05:44 +04:00 · 2021-11-20 16:05:44 +04:00 · 388964e8d0
commit 388964e8d0
parent 4b6bf003b8
4 changed files with 4 additions and 4 deletions
--- a/models/Model_AMP/Model.py
+++ b/models/Model_AMP/Model.py
@ -625,7 +625,7 @@ class AMPModel(ModelBase):
                name='AMP',
                input_names=['in_face:0','morph_value:0'],
                output_names=['out_face_mask:0','out_celeb_face:0','out_celeb_face_mask:0'],
-                opset=9,
+                opset=12,
                output_path=output_path)

    #override
--- a/models/Model_SAEHD/Model.py
+++ b/models/Model_SAEHD/Model.py
@ -746,7 +746,7 @@ Examples: df, liae, df-d, df-ud, liae-ud, ...
                name='SAEHD',
                input_names=['in_face:0'],
                output_names=['out_face_mask:0','out_celeb_face:0','out_celeb_face_mask:0'],
-                opset=9,
+                opset=12,
                output_path=output_path)

    #override
--- a/requirements-colab.txt
+++ b/requirements-colab.txt
@ -8,4 +8,4 @@ scikit-image==0.14.2
 scipy==1.4.1
 colorama
 tensorflow-gpu==2.4.0
-tf2onnx==1.8.4
+tf2onnx==1.9.2
--- a/requirements-cuda.txt
+++ b/requirements-cuda.txt
@ -9,4 +9,4 @@ scipy==1.4.1
 colorama
 tensorflow-gpu==2.4.0
 pyqt5
-tf2onnx==1.8.4
+tf2onnx==1.9.2