[Enhance] Improve control of 3D visualisation of pose lifter (#1481)

* Add some missing parameter documentation

* Expose vis_height / axis_azimuth

* Fix interhand axis_azimuth

* keep original default behavior

Co-authored-by: ly015 <liyining0712@gmail.com>

demo/interhand3d_img_demo.py

@@ -251,6 +251,7 @@ def main():
             kpt_score_thr=args.kpt_thr,
             radius=args.radius,
             thickness=args.thickness,
+            axis_azimuth=-115,
         )
 
 

mmpose/apis/inference_3d.py

@@ -364,7 +364,9 @@ def vis_3d_pose_result(model,
                        kpt_score_thr=0.3,
                        radius=8,
                        thickness=2,
+                       vis_height=400,
                        num_instances=-1,
+                       axis_azimuth=70,
                        show=False,
                        out_file=None):
     """Visualize the 3D pose estimation results.
@@ -459,7 +461,9 @@ def vis_3d_pose_result(model,
         thickness=thickness,
         pose_kpt_color=pose_kpt_color,
         pose_link_color=pose_link_color,
+        vis_height=vis_height,
         num_instances=num_instances,
+        axis_azimuth=axis_azimuth,
         show=show,
         out_file=out_file)
 

mmpose/models/detectors/interhand_3d.py

@@ -116,6 +116,7 @@ def show_result(self,
                     pose_link_color=None,
                     vis_height=400,
                     num_instances=-1,
+                    axis_azimuth=-115,
                     win_name='',
                     show=False,
                     wait_time=0,
@@ -150,6 +151,7 @@ def show_result(self,
                 smaller than 0, all the instances in the pose_result will be
                 shown. Otherwise, pad or truncate the pose_result to a length
                 of num_instances.
+            axis_azimuth (float): axis azimuth angle for 3D visualizations.
             win_name (str): The window name.
             show (bool): Whether to show the image. Default: False.
             wait_time (int): Value of waitKey param.
@@ -213,7 +215,7 @@ def show_result(self,
             pose_link_color,
             vis_height,
             axis_limit=300,
-            axis_azimuth=-115,
+            axis_azimuth=axis_azimuth,
             axis_elev=15,
             kpt_score_thr=kpt_score_thr,
             num_instances=num_instances)

mmpose/models/detectors/pose_lifter.py

@@ -296,6 +296,7 @@ def show_result(self,
                     thickness=2,
                     vis_height=400,
                     num_instances=-1,
+                    axis_azimuth=70,
                     win_name='',
                     show=False,
                     wait_time=0,
@@ -323,7 +324,13 @@ def show_result(self,
             vis_height (int): The image height of the visualization. The width
                 will be N*vis_height depending on the number of visualized
                 items.
+            num_instances (int): Number of instances to be shown in 3D. If
+                smaller than 0, all the instances in the result will be shown.
+                Otherwise, pad or truncate the result to a length of
+                num_instances.
+            axis_azimuth (float): axis azimuth angle for 3D visualizations.
             win_name (str): The window name.
+            show (bool): Whether to directly show the visualization.
             wait_time (int): Value of waitKey param.
                 Default: 0.
             out_file (str or None): The filename to write the image.
@@ -383,7 +390,9 @@ def show_result(self,
             pose_kpt_color,
             pose_link_color,
             vis_height,
-            num_instances=num_instances)
+            num_instances=num_instances,
+            axis_azimuth=axis_azimuth,
+        )
 
         if show:
             mmcv.visualization.imshow(img_vis, win_name, wait_time)