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

initial code to extract umdfaces.io dataset and train pose estimator

Browse source
This commit is contained in:
iperov 2019-04-23 08:14:09 +04:00
parent 51a917facc
commit e58197ca22
18 changed files with 437 additions and 57 deletions
Download patch file Download diff file Expand all files Collapse all files

2
facelib/FANSegmentator.py
View file

@ -26,7 +26,7 @@ class FANSegmentator(object):
self.model = FANSegmentator.BuildModel(resolution, ngf=64)
if weights_file_root:
if weights_file_root is not None:
weights_file_root = Path(weights_file_root)
else:
weights_file_root = Path(__file__).parent