Change saved result format (#5698)

* change id to image path when save result of image directory prediction

* fix error info

deploy/python/det_keypoint_unite_infer.py

@@ -85,8 +85,9 @@ def topdown_unite_predict(detector,
                 FLAGS.run_benchmark)
 
             if save_res:
+                save_name = img_file if isinstance(img_file, str) else i
                 store_res.append([
-                    i, keypoint_res['bbox'],
+                    save_name, keypoint_res['bbox'],
                     [keypoint_res['keypoint'][0], keypoint_res['keypoint'][1]]
                 ])
         else:

deploy/python/infer.py

@@ -691,7 +691,7 @@ def get_test_images(infer_dir, infer_img):
     Get image path list in TEST mode
     """
     assert infer_img is not None or infer_dir is not None, \
-        "--infer_img or --infer_dir should be set"
+        "--image_file or --image_dir should be set"
     assert infer_img is None or os.path.isfile(infer_img), \
             "{} is not a file".format(infer_img)
     assert infer_dir is None or os.path.isdir(infer_dir), \

deploy/python/mot_keypoint_unite_infer.py

@@ -95,8 +95,9 @@ def mot_topdown_unite_predict(mot_detector,
             FLAGS.run_benchmark)
 
         if save_res:
+            save_name = img_file if isinstance(img_file, str) else i
             store_res.append([
-                i, keypoint_res['bbox'],
+                save_name, keypoint_res['bbox'],
                 [keypoint_res['keypoint'][0], keypoint_res['keypoint'][1]]
             ])
         if FLAGS.run_benchmark: