diff --git a/deploy/python/keypoint_infer.py b/deploy/python/keypoint_infer.py
index b87bac92c3958c23c392e0d43a688a1346c9744c..4227a281aa5e7ae664e32d8272c800be3ba247f9 100644
--- a/deploy/python/keypoint_infer.py
+++ b/deploy/python/keypoint_infer.py
@@ -267,8 +267,10 @@ class KeyPointDetector(Detector):
             print('detect frame: %d' % (index))
             index += 1
             results = self.predict_image([frame], visual=False)
+            im_results = {}
+            im_results['keypoint'] = [results['keypoint'], results['score']]
             im = visualize_pose(
-                frame, results, visual_thresh=self.threshold, returnimg=True)
+                frame, im_results, visual_thresh=self.threshold, returnimg=True)
             writer.write(im)
             if camera_id != -1:
                 cv2.imshow('Mask Detection', im)