update scripts

dboyliao · dboyliao · commit 36e12614e2db · 2019-03-23T15:40:46.000+08:00
diff --git a/prepare_test_data.py b/prepare_test_data.py
@@ -51,6 +51,7 @@ def prepare_test_data(accept_prob, labels):
             fid.write("    },\n")
         fid.write("};\n")
         fid.write("#endif // __IMG_DATA_H\n")
+    click.secho("img_data.h generated", fg='yellow', bold=True)
 
 
 if __name__ == "__main__":
