StarMirror/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2024-03-22 13:10:55 +08:00
Code Issues Projects Releases Wiki Activity

fix model export. Update tf2onnx to 1.9.2

Browse Source
This commit is contained in:
iperov 2021-11-20 16:05:44 +04:00
parent 4b6bf003b8
commit 388964e8d0
4 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
models/Model_AMP/Model.py
View File

@ -625,7 +625,7 @@ class AMPModel(ModelBase):
name='AMP', name='AMP',
input_names=['in_face:0','morph_value:0'], input_names=['in_face:0','morph_value:0'],
output_names=['out_face_mask:0','out_celeb_face:0','out_celeb_face_mask:0'], output_names=['out_face_mask:0','out_celeb_face:0','out_celeb_face_mask:0'],
opset=9, opset=12,
output_path=output_path) output_path=output_path)
#override #override

2
models/Model_SAEHD/Model.py
View File

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

2
requirements-colab.txt
View File

@ -8,4 +8,4 @@ scikit-image==0.14.2
scipy==1.4.1 scipy==1.4.1
colorama colorama
tensorflow-gpu==2.4.0 tensorflow-gpu==2.4.0
tf2onnx==1.8.4 tf2onnx==1.9.2

2
requirements-cuda.txt
View File

@ -9,4 +9,4 @@ scipy==1.4.1
colorama colorama
tensorflow-gpu==2.4.0 tensorflow-gpu==2.4.0
pyqt5 pyqt5
tf2onnx==1.8.4 tf2onnx==1.9.2