fix visualization bugs in mono3d (#1066)

mmdet3d/core/visualizer/show_result.py

@@ -132,7 +132,7 @@ def show_seg_result(points,
                     filename,
                     palette,
                     ignore_index=None,
-                    show=True,
+                    show=False,
                     snapshot=False):
     """Convert results into format that is directly readable for meshlab.
 
@@ -206,7 +206,7 @@ def show_multi_modality_result(img,
                                filename,
                                box_mode='lidar',
                                img_metas=None,
-                               show=True,
+                               show=False,
                                gt_bbox_color=(61, 102, 255),
                                pred_bbox_color=(241, 101, 72)):
     """Convert multi-modality detection results into 2D results.

mmdet3d/datasets/kitti_mono_dataset.py

@@ -202,7 +202,8 @@ class KittiMonoDataset(NuScenesMonoDataset):
                  pklfile_prefix=None,
                  submission_prefix=None,
                  show=False,
-                 out_dir=None):
+                 out_dir=None,
+                 pipeline=None):
         """Evaluation in KITTI protocol.
 
         Args:
@@ -220,6 +221,8 @@ class KittiMonoDataset(NuScenesMonoDataset):
                 Default: False.
             out_dir (str, optional): Path to save the visualization results.
                 Default: None.
+            pipeline (list[dict], optional): raw data loading for showing.
+                Default: None.
 
         Returns:
             dict[str, float]: Results of each evaluation metric.
@@ -256,8 +259,8 @@ class KittiMonoDataset(NuScenesMonoDataset):
 
         if tmp_dir is not None:
             tmp_dir.cleanup()
-        if show:
-            self.show(results, out_dir)
+        if show or out_dir:
+            self.show(results, out_dir, show=show, pipeline=pipeline)
         return ap_dict
 
     def bbox2result_kitti(self,

mmdet3d/datasets/utils.py

@@ -12,7 +12,7 @@ from mmdet3d.datasets.pipelines import (Collect3D, DefaultFormatBundle3D,
                                         PointSegClassMapping)
 # yapf: enable
 from mmdet.datasets.builder import PIPELINES
-from mmdet.datasets.pipelines import LoadImageFromFile
+from mmdet.datasets.pipelines import LoadImageFromFile, MultiScaleFlipAug
 
 
 def is_loading_function(transform):
@@ -40,12 +40,12 @@ def is_loading_function(transform):
             return False
         if obj_cls in loading_functions:
             return True
-        if obj_cls in (MultiScaleFlipAug3D, ):
+        if obj_cls in (MultiScaleFlipAug3D, MultiScaleFlipAug):
             return None
     elif callable(transform):
         if isinstance(transform, loading_functions):
             return True
-        if isinstance(transform, MultiScaleFlipAug3D):
+        if isinstance(transform, (MultiScaleFlipAug3D, MultiScaleFlipAug)):
             return None
     return False