Merge 9dc3a42 into 4f35db0

Author: Ben-Louis

demo/body3d_pose_lifter_demo.py

@@ -1,4 +1,5 @@
 # Copyright (c) OpenMMLab. All rights reserved.
+import logging
 import mimetypes
 import os
 import time
@@ -10,6 +11,7 @@
 import mmcv
 import mmengine
 import numpy as np
+from mmengine.logging import print_log
 from mmengine.structures import InstanceData
 
 from mmpose.apis import (_track_by_iou, _track_by_oks, collect_multi_frames,
@@ -477,6 +479,13 @@ def main():
                 indent='\t')
         print(f'predictions have been saved at {args.pred_save_path}')
 
+    if save_output:
+        input_type = input_type.replace('webcam', 'video')
+        print_log(
+            f'the output {input_type} has been saved at {output_file}',
+            logger='current',
+            level=logging.INFO)
+
 
 if __name__ == '__main__':
     main()

demo/bottomup_demo.py

@@ -1,4 +1,5 @@
 # Copyright (c) OpenMMLab. All rights reserved.
+import logging
 import mimetypes
 import os
 import time
@@ -9,6 +10,7 @@
 import mmcv
 import mmengine
 import numpy as np
+from mmengine.logging import print_log
 
 from mmpose.apis import inference_bottomup, init_model
 from mmpose.registry import VISUALIZERS
@@ -223,6 +225,13 @@ def main():
                 indent='\t')
         print(f'predictions have been saved at {args.pred_save_path}')
 
+    if output_file:
+        input_type = input_type.replace('webcam', 'video')
+        print_log(
+            f'the output {input_type} has been saved at {output_file}',
+            logger='current',
+            level=logging.INFO)
+
 
 if __name__ == '__main__':
     main()

demo/image_demo.py

@@ -1,7 +1,9 @@
 # Copyright (c) OpenMMLab. All rights reserved.
+import logging
 from argparse import ArgumentParser
 
 from mmcv.image import imread
+from mmengine.logging import print_log
 
 from mmpose.apis import inference_topdown, init_model
 from mmpose.registry import VISUALIZERS
@@ -100,6 +102,12 @@ def main():
         show=args.show,
         out_file=args.out_file)
 
+    if args.out_file is not None:
+        print_log(
+            f'the output image has been saved at {args.out_file}',
+            logger='current',
+            level=logging.INFO)
+
 
 if __name__ == '__main__':
     main()

demo/topdown_demo_with_mmdet.py

@@ -1,4 +1,5 @@
 # Copyright (c) OpenMMLab. All rights reserved.
+import logging
 import mimetypes
 import os
 import time
@@ -9,6 +10,7 @@
 import mmcv
 import mmengine
 import numpy as np
+from mmengine.logging import print_log
 
 from mmpose.apis import inference_topdown
 from mmpose.apis import init_model as init_pose_estimator
@@ -288,6 +290,13 @@ def main():
                 indent='\t')
         print(f'predictions have been saved at {args.pred_save_path}')
 
+    if output_file:
+        input_type = input_type.replace('webcam', 'video')
+        print_log(
+            f'the output {input_type} has been saved at {output_file}',
+            logger='current',
+            level=logging.INFO)
+
 
 if __name__ == '__main__':
     main()

mmpose/apis/inferencers/base_mmpose_inferencer.py

@@ -1,4 +1,5 @@
 # Copyright (c) OpenMMLab. All rights reserved.
+import logging
 import mimetypes
 import os
 import warnings
@@ -16,6 +17,7 @@
 from mmengine.fileio import (get_file_backend, isdir, join_path,
                              list_dir_or_file)
 from mmengine.infer.infer import BaseInferencer
+from mmengine.logging import print_log
 from mmengine.registry import init_default_scope
 from mmengine.runner.checkpoint import _load_checkpoint_to_model
 from mmengine.structures import InstanceData
@@ -358,6 +360,7 @@ def visualize(self,
                             file_name = os.path.basename(
                                 self.video_info['name'])
                         out_file = join_path(dir_name, file_name)
+                        self.video_info['output_file'] = out_file
                         self.video_info['writer'] = cv2.VideoWriter(
                             out_file, fourcc, self.video_info['fps'],
                             (visualization.shape[1], visualization.shape[0]))
@@ -367,6 +370,10 @@ def visualize(self,
                     file_name = file_name if file_name else img_name
                     out_file = join_path(dir_name, file_name)
                     mmcv.imwrite(out_img, out_file)
+                    print_log(
+                        f'the output image has been saved at {out_file}',
+                        logger='current',
+                        level=logging.INFO)
 
         if return_vis:
             return results
@@ -454,6 +461,11 @@ def _finalize_video_processing(
 
         # Release the video writer if it exists
         if self.video_info['writer'] is not None:
+            out_file = self.video_info['output_file']
+            print_log(
+                f'the output video has been saved at {out_file}',
+                logger='current',
+                level=logging.INFO)
             self.video_info['writer'].release()
 
         # Save predictions

mmpose/apis/inferencers/pose3d_inferencer.py

@@ -1,4 +1,5 @@
 # Copyright (c) OpenMMLab. All rights reserved.
+import logging
 import os
 import warnings
 from collections import defaultdict
@@ -12,6 +13,7 @@
 from mmengine.config import Config, ConfigDict
 from mmengine.fileio import join_path
 from mmengine.infer.infer import ModelType
+from mmengine.logging import print_log
 from mmengine.model import revert_sync_batchnorm
 from mmengine.registry import init_default_scope
 from mmengine.structures import InstanceData
@@ -509,6 +511,7 @@ def visualize(self,
                             file_name = os.path.basename(
                                 self.video_info['name'])
                         out_file = join_path(dir_name, file_name)
+                        self.video_info['output_file'] = out_file
                         self.video_info['writer'] = cv2.VideoWriter(
                             out_file, fourcc, self.video_info['fps'],
                             (visualization.shape[1], visualization.shape[0]))
@@ -519,6 +522,10 @@ def visualize(self,
                     file_name = file_name if file_name else img_name
                     out_file = join_path(dir_name, file_name)
                     mmcv.imwrite(out_img, out_file)
+                    print_log(
+                        f'the output image has been saved at {out_file}',
+                        logger='current',
+                        level=logging.INFO)
 
         if return_vis:
             return results