test script

opencv-yolo/web/test_videoai.py

@@ -38,7 +38,7 @@ class VideoAiTestor:
     if key in self.videos:
       self.cnt -= 1
       self.results[key] = v
-      print("got result for {}: {}".format(key, v))
+      print("left {}, got result for {}: {}".format(self.cnt, key, v))
   
   def __init__(self,cfg_path):
     self.cfg_path = cfg_path
@@ -51,6 +51,9 @@ class VideoAiTestor:
         if not self.TASK_VALIDATOR.validate(v):
           raise
         key = v['cameraId'] + str(v['startTime'])
+        if key in self.videos:
+          print("duplicated video json, ignored: {}".format(json.dumps(v)))
+          continue
         self.videos[key] = v
         self.cnt += 1
     print("loaded {} lines of videos json".format(self.cnt))
@@ -62,6 +65,7 @@ class VideoAiTestor:
 
   def run(self):
     mc = mqtt.Client("vatstpub")
+    mc.max_inflight_messages_set(self.cnt)
     mc.connect(self.MQTT_HOST, self.MQTT_PORT)
     for k, v in self.videos.items():
       msg = json.dumps(v)