Fix pointrend missing get_uncertainty function bug (open-mmlab#7550)

* [Fix] Adjust the order of get_classes and FileClient. (open-mmlab#7276)

* delete -sv (open-mmlab#7277)

Co-authored-by: Wenwei Zhang <[email protected]>

* [Docs] Add Chinese version of finetune (open-mmlab#7178)

* [Fix] Fix wrong img name in onnx2tensorrt.py (open-mmlab#7157)

* [Docs] fix albumentations installed way (open-mmlab#7143)

* Update finetune.md

Translate the finetune.md doc to Chinese

* Update finetune.md

* Update finetune.md

* Update finetune.md

* fix lint

* fx lint

* fix pr

Co-authored-by: Jamie <[email protected]>
Co-authored-by: BigDong <[email protected]>

* set unmap_results=True in ssd_head (open-mmlab#7328)

* Update YOLOX log for non square input (open-mmlab#7235)

* [Enhance] add cpu_num in cocopanoptic for pq computing (open-mmlab#7315)

* add cpu_num in cocopanoptic for pq computing

* cpu_num -> nproc

* move nproc to evaluate

* [Enhancement] Allow to set channel_order in LoadImageFromFile (open-mmlab#7258)

* allow to set channel_order when loading images

* fix lint

* fix unit test

* fix lint

* [Fix] Force the inputs of `get_bboxes` in yolox_head to float32. (open-mmlab#7324)

* Fix softnms bug

* Add force_fp32 in corner_head and centripetal_head

* [Fix] Fix typo in FPN neck (open-mmlab#7347)

* update readme and pretrained related (open-mmlab#7301)

* [Docs] Add Chinese version of onnx2tensorrt.md (open-mmlab#7219)

* Fix bug of docs

* translate onnx2tensorrt.md

* fix

* fix end-of-file-fixer

* fix some bugs

* 修复链接跳转

* 修复链接跳转

* 修复链接跳转-测试1

* 修复链接跳转-测试2

* 修复链接跳转-测试2

* 修复链接跳转-测试3

* 修复链接跳转-测试5

* Fix

Co-authored-by: jbwang1997 <[email protected]>

* Update useful_tools.md (open-mmlab#7180)

* [Enhancement]: Update colab tutorials (open-mmlab#7310)

* update colab tutorials

* update

* fix

* fix wrong CUDA explaination

* resolve comments

* resolve comments

* fix typo

Co-authored-by: Cedric Luo <[email protected]>
Co-authored-by: tripleMu <[email protected]>
Co-authored-by: jbwang1997 <[email protected]>
Co-authored-by: kira <[email protected]>
Co-authored-by: Wenwei Zhang <[email protected]>

* Fix pointrend missing get_uncertainty function bug

Co-authored-by: Wencheng Wu <[email protected]>
Co-authored-by: Yue Zhou <[email protected]>
Co-authored-by: Wenwei Zhang <[email protected]>
Co-authored-by: MingJian.L <[email protected]>
Co-authored-by: Jamie <[email protected]>
Co-authored-by: BigDong <[email protected]>
Co-authored-by: Cedric Luo <[email protected]>
Co-authored-by: Yosuke Shinya <[email protected]>
Co-authored-by: Cedric Luo <[email protected]>
Co-authored-by: Jingwei Zhang <[email protected]>
Co-authored-by: jbwang1997 <[email protected]>
Co-authored-by: Xiangxu-0103 <[email protected]>
Co-authored-by: tripleMu <[email protected]>
Co-authored-by: kira <[email protected]>

mmdet/models/roi_heads/mask_heads/mask_point_head.py

@@ -8,7 +8,8 @@
 from mmcv.runner import BaseModule
 
 from mmdet.models.builder import HEADS, build_loss
-from mmdet.models.utils import get_uncertain_point_coords_with_randomness
+from mmdet.models.utils import (get_uncertain_point_coords_with_randomness,
+                                get_uncertainty)
 
 
 @HEADS.register_module()
@@ -230,7 +231,7 @@ def get_roi_rel_points_test(self, mask_pred, pred_label, cfg):
                 most uncertain points from the [mask_height, mask_width] grid .
         """
         num_points = cfg.subdivision_num_points
-        uncertainty_map = self._get_uncertainty(mask_pred, pred_label)
+        uncertainty_map = get_uncertainty(mask_pred, pred_label)
         num_rois, _, mask_height, mask_width = uncertainty_map.shape
 
         # During ONNX exporting, the type of each elements of 'shape' is

mmdet/models/utils/__init__.py

@@ -10,7 +10,8 @@
 from .misc import interpolate_as, sigmoid_geometric_mean
 from .normed_predictor import NormedConv2d, NormedLinear
 from .panoptic_gt_processing import preprocess_panoptic_gt
-from .point_sample import get_uncertain_point_coords_with_randomness
+from .point_sample import (get_uncertain_point_coords_with_randomness,
+                           get_uncertainty)
 from .positional_encoding import (LearnedPositionalEncoding,
                                   SinePositionalEncoding)
 from .res_layer import ResLayer, SimplifiedBasicBlock
@@ -29,5 +30,5 @@
     'adaptive_avg_pool2d', 'AdaptiveAvgPool2d', 'PatchEmbed', 'nchw_to_nlc',
     'nlc_to_nchw', 'pvt_convert', 'sigmoid_geometric_mean',
     'preprocess_panoptic_gt', 'DyReLU',
-    'get_uncertain_point_coords_with_randomness'
+    'get_uncertain_point_coords_with_randomness', 'get_uncertainty'
 ]