add comments

open-mmlab · Feb 15, 2023 · 8b8cf9e · 8b8cf9e
1 parent 5c50239
commit 8b8cf9e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/mmdet3d/datasets/transforms/formating.py b/mmdet3d/datasets/transforms/formating.py
@@ -155,6 +155,10 @@ def pack_single_results(self, results: dict) -> dict:
                 img = results['img']
                 if len(img.shape) < 3:
                     img = np.expand_dims(img, -1)
+                # To improve the computational speed by by 3-5 times, apply:
+                # `torch.permute()` rather than `np.transpose()`.
+                # Refer to https://github.com/open-mmlab/mmdetection/pull/9533
+                # for more details
                 results['img'] = to_tensor(img).permute(2, 0, 1).contiguous()
 
         for key in [