github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-08 05:51:40 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

_

Browse source
This commit is contained in:
Colombo 2020-03-09 10:15:29 +04:00
parent f07349cc32
commit 9eb15065f9
1 changed files with 0 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
models/Model_FANSeg/Model.py
View file

@ -172,7 +172,6 @@ class FANSegModel(ModelBase):
st = [] st = []
for i in range(n_samples): for i in range(n_samples):
ar = S[i]*TM[i]+ green_bg*(1-TM[i]), SM[i], S[i]*SM[i] + green_bg*(1-SM[i]) ar = S[i]*TM[i]+ green_bg*(1-TM[i]), SM[i], S[i]*SM[i] + green_bg*(1-SM[i])
#todo green bg
st.append ( np.concatenate ( ar, axis=1) ) st.append ( np.concatenate ( ar, axis=1) )
result += [ ('FANSeg training faces', np.concatenate (st, axis=0 )), ] result += [ ('FANSeg training faces', np.concatenate (st, axis=0 )), ]
@ -185,7 +184,6 @@ class FANSegModel(ModelBase):
st = [] st = []
for i in range(n_samples): for i in range(n_samples):
ar = D[i], DM[i], D[i]*DM[i]+ green_bg*(1-DM[i]) ar = D[i], DM[i], D[i]*DM[i]+ green_bg*(1-DM[i])
#todo green bg
st.append ( np.concatenate ( ar, axis=1) ) st.append ( np.concatenate ( ar, axis=1) )
result += [ ('FANSeg unseen faces', np.concatenate (st, axis=0 )), ] result += [ ('FANSeg unseen faces', np.concatenate (st, axis=0 )), ]