From 2ed6d59b99db6bf9d9bce00be22f8aa3b5fe0e8b Mon Sep 17 00:00:00 2001
From: nttstar <guojia@gmail.com>
Date: Thu, 8 Feb 2018 19:36:54 +0800
Subject: [PATCH] fix

---
 src/data.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/data.py b/src/data.py
index 4ead151..65b037d 100644
--- a/src/data.py
+++ b/src/data.py
@@ -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: