diff --git a/deploy/python/det_keypoint_unite_infer.py b/deploy/python/det_keypoint_unite_infer.py
index 2b901dcaa9dd7af8fcff4dcc99e15acd4ee9dede..c3295559778e2a7c61a68e36cb3971cb3e83f7f7 100644
--- a/deploy/python/det_keypoint_unite_infer.py
+++ b/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:
diff --git a/deploy/python/infer.py b/deploy/python/infer.py
index d712ed7e82477e0ed7e51f346eaaa1ad05866abc..c8737ccfdb5a87364ca9a4c250dd0204df5798fc 100644
--- a/deploy/python/infer.py
+++ b/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), \
diff --git a/deploy/python/mot_keypoint_unite_infer.py b/deploy/python/mot_keypoint_unite_infer.py
index 96ed8e399f501cc49ccc2fbcb5109175e90ab755..9f8413bb60b17d69e46ed21f189785ea461cf8a5 100644
--- a/deploy/python/mot_keypoint_unite_infer.py
+++ b/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: