save results to a json file

Author: foamliu

demo.py

@@ -1,4 +1,5 @@
 # import the necessary packages
+import json
 import os
 import random
 
@@ -39,4 +40,8 @@
         results.append({'label': class_names[class_id][0][0], 'prob': '{:.4}'.format(prob)})
         cv.imwrite('images/{}_out.png'.format(i), bgr_img)
 
+    print(results)
+    with open('results.json', 'w') as file:
+        json.dump(results, file)
+
     K.clear_session()