diff --git a/deploy/python/keypoint_det_unite_infer.py b/deploy/python/keypoint_det_unite_infer.py
index dd80b290ff89377562dfb7149ed05a9257fd079b..88ae6cf07cdac497a0f2faa9d7c7840a233b137f 100644
--- a/deploy/python/keypoint_det_unite_infer.py
+++ b/deploy/python/keypoint_det_unite_infer.py
@@ -68,7 +68,7 @@ def affine_backto_orgimages(keypoint_result, batch_records):
 def topdown_unite_predict(detector, topdown_keypoint_detector, image_list):
     for i, img_file in enumerate(image_list):
         image, _ = decode_image(img_file, {})
-        results = detector.predict(image, FLAGS.det_threshold)
+        results = detector.predict([image], FLAGS.det_threshold)
         batchs_images, det_rects = get_person_from_rect(image, results)
         keypoint_vector = []
         score_vector = []
@@ -121,7 +121,7 @@ def topdown_unite_predict_video(detector, topdown_keypoint_detector, camera_id):
         print('detect frame:%d' % (index))
 
         frame2 = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
-        results = detector.predict(frame2, FLAGS.det_threshold)
+        results = detector.predict([frame2], FLAGS.det_threshold)
         batchs_images, rect_vecotr = get_person_from_rect(frame2, results)
         keypoint_vector = []
         score_vector = []