diff --git a/models/Model_AMP/Model.py b/models/Model_AMP/Model.py index 54d3144..82b0dc5 100644 --- 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 diff --git a/models/Model_SAEHD/Model.py b/models/Model_SAEHD/Model.py index 74fd5c1..ecfaa73 100644 --- 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 diff --git a/requirements-colab.txt b/requirements-colab.txt index 8aeefe5..ad52e3a 100644 --- 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 \ No newline at end of file +tf2onnx==1.9.2 \ No newline at end of file diff --git a/requirements-cuda.txt b/requirements-cuda.txt index 6f12970..888eeae 100644 --- 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 \ No newline at end of file +tf2onnx==1.9.2 \ No newline at end of file