bugfixed: reference before assignment for img_shape in KittiDataset

open-mmlab · Mar 19, 2020 · 826faec · 826faec
1 parent 91c321a
commit 826faec
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pcdet/datasets/kitti/kitti_dataset.py b/pcdet/datasets/kitti/kitti_dataset.py
@@ -400,8 +400,8 @@ def __getitem__(self, index):
         points = self.get_lidar(sample_idx)
         calib = self.get_calib(sample_idx)
 
+        img_shape = info['image']['image_shape']
         if cfg.DATA_CONFIG.FOV_POINTS_ONLY:
-            img_shape = info['image']['image_shape']
             pts_rect = calib.lidar_to_rect(points[:, 0:3])
             fov_flag = self.get_fov_flag(pts_rect, img_shape, calib)
             points = points[fov_flag]