加速label转color

coderzbx · zhangbenxing · commit 641d26a0f5bb · 2017-11-23T10:23:03.000+08:00
diff --git a/SelfScripts/demo.py b/SelfScripts/demo.py
@@ -58,6 +58,8 @@ def do(self, image_data):
 
 if __name__ == '__main__':
 
+    time1 = time.time()
+
     weights = ''
     model = ''
     colours = ''
@@ -176,3 +178,8 @@ def do(self, image_data):
 
                 with open(recog_path, 'wb') as w:
                     w.write(recog_data)
+
+    time2 = time.time()
+
+    print("finish in {} s\n".format(time2 - time1))
+
