ZF/insightface
1
0
Fork 0
mirror of https://github.com/deepinsight/insightface.git synced 2026-05-15 12:52:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
nttstar
2018-02-08 19:36:54 +08:00
parent f1cd542e0f
commit 2ed6d59b99
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/data.py
View File

@@ -77,7 +77,7 @@ class FaceImageIter(io.DataIter):
self.idx2flag = {}
self.idx2meancos = {}
self.c2c_auto = False
if output_c2c or c2c_threshold>0.0:
if output_c2c or c2c_threshold>0.0 or c2c_mode>=-5:
path_c2c = os.path.join(os.path.dirname(path_imgrec), 'c2c')
print(path_c2c)
if os.path.exists(path_c2c):
@@ -745,7 +745,7 @@ class FaceImageIter(io.DataIter):
meancos = self.idx2meancos[idx]
label = [label, meancos]
else:
if isinstance(label, list):
if not isinstance(label, numbers.Number):
label = label[0]
return label, img, None, None
else: