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

2019-04-23 08:14:09 +04:00
parent 51a917facc
commit e58197ca22
18 changed files with 437 additions and 57 deletions
@@ -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