From b20a2e290219b6ba4f2e18bb80e87f573448005c Mon Sep 17 00:00:00 2001
From: Francesco Gatti <gattifrancesco@hotmail.it>
Date: Fri, 4 Aug 2017 10:45:07 +0200
Subject: [PATCH] fix

---
 include/utils.h     | 2 +-
 tests/yolo/yolo.cpp | 7 +++----
 2 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/include/utils.h b/include/utils.h
index e724dc2..7c10f0f 100644
--- a/include/utils.h
+++ b/include/utils.h
@@ -21,7 +21,7 @@
 #define TIMER_STOP  clock_gettime(CLOCK_MONOTONIC, &end);              \
     double t_ns = ((double)(end.tv_sec - start.tv_sec) * 1.0e9 +       \
                   (double)(end.tv_nsec - start.tv_nsec))/1.0e6;        \
-    std::cout<<"Time:"<<std::setw(16)<<t_ns<<" ms\n"; 
+    std::cout<<"\033[1;36mTime:"<<std::setw(16)<<t_ns<<" ms\033[0m\n"; 
 
 
 /********************************************************
diff --git a/tests/yolo/yolo.cpp b/tests/yolo/yolo.cpp
index 99ae44b..fedd707 100644
--- a/tests/yolo/yolo.cpp
+++ b/tests/yolo/yolo.cpp
@@ -129,9 +129,8 @@ int main() {
     value_type *out, *out_h;
     int out_dim = net.getOutputDim().tot();
     readBinaryFile(output_bin, out_dim, &out_h, &out);
-    std::cout<<"CUDNN vs correct Diffs: "<<checkResult(out_dim, out_data, out)<<"\n";
-    std::cout<<"TRT   vs correct Diffs: "<<checkResult(out_dim, out_data2, out)<<"\n";
-    std::cout<<"CUDNN vs TRT     Diffs: "<<checkResult(out_dim, out_data, out)<<"\n";
-
+    std::cout<<"CUDNN vs correct Wrongs: "<<checkResult(out_dim, out_data, out)<<"\n";
+    std::cout<<"TRT   vs correct Wrongs: "<<checkResult(out_dim, out_data2, out)<<"\n";
+    std::cout<<"CUDNN vs TRT     Wrongs: "<<checkResult(out_dim, out_data, out)<<"\n";
     return 0;
 }