Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Fix pointrend missing get_uncertainty function bug #7550

Merged
merged 12 commits into from
Mar 28, 2022
Merged

Fix pointrend missing get_uncertainty function bug #7550

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
fa4d608
[Fix] Adjust the order of get_classes and FileClient. (#7276)
274869388 Mar 1, 2022
86c2b4d
delete -sv (#7277)
zytx121 Mar 1, 2022
93cf809
[Docs] Add Chinese version of finetune (#7178)
matrixgame2018 Mar 3, 2022
83b8d72
set unmap_results=True in ssd_head (#7328)
chhluo Mar 6, 2022
4d6f2c4
Update YOLOX log for non square input (#7235)
shinya7y Mar 6, 2022
0ea978f
[Enhance] add cpu_num in cocopanoptic for pq computing (#7315)
chhluo Mar 7, 2022
ac7bfb3
[Enhancement] Allow to set channel_order in LoadImageFromFile (#7258)
JingweiZhang12 Mar 9, 2022
c0a5005
[Fix] Force the inputs of `get_bboxes` in yolox_head to float32. (#7324)
jbwang1997 Mar 9, 2022
5692393
[Fix] Fix typo in FPN neck (#7347)
Xiangxu-0103 Mar 9, 2022
69fc0ad
Merge branch 'dev' of https://github.com/open-mmlab/mmdetection into dev
hhaAndroid Mar 21, 2022
8c01ce6
Merge branch 'dev' of https://github.com/open-mmlab/mmdetection into dev
hhaAndroid Mar 28, 2022
dac0187
Fix pointrend missing get_uncertainty function bug
hhaAndroid Mar 28, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 3 additions & 2 deletions mmdet/models/roi_heads/mask_heads/mask_point_head.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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()
Expand Down Expand Up @@ -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
Expand Down
5 changes: 3 additions & 2 deletions mmdet/models/utils/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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'
]