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.

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

Refactor one-stage get_bboxes logic #5317

Merged
merged 73 commits into from
Aug 11, 2021
Merged
Changes from 1 commit
Commits
Show all changes
73 commits
Select commit Hold shift + click to select a range
77da249
revert batch to single
hhaAndroid Jun 7, 2021
2ad29eb
update anchor_head
hhaAndroid Jun 8, 2021
2db854a
replace preds with bboxes
hhaAndroid Jun 8, 2021
060b1c7
add point_bbox_coder
hhaAndroid Jun 8, 2021
172d37a
FCOS add get_selected_priori
hhaAndroid Jun 8, 2021
dd9e514
unified anchor-free and anchor-based get_bbox_single
hhaAndroid Jun 8, 2021
64912e2
update code
hhaAndroid Jun 11, 2021
91c8d46
update reppoints and sabl
hhaAndroid Jun 11, 2021
a135f18
add sparse priors
jshilong Jun 14, 2021
63364cd
add mlvlpointsgenerator
jshilong Jun 14, 2021
e084958
revert __init__ of core
jshilong Jun 14, 2021
706379c
refactor reppoints
jshilong Jun 14, 2021
566b973
delete label channal
jshilong Jun 15, 2021
6f9d973
add docstr
jshilong Jun 15, 2021
f3aafe0
fix typo
jshilong Jun 15, 2021
18a0fdc
fix args
jshilong Jun 15, 2021
f29b5a9
fix typo
jshilong Jun 15, 2021
d8b121f
fix doc
jshilong Jun 16, 2021
07e182e
fix stride_h
jshilong Jun 16, 2021
6c8f5f5
add offset
jshilong Jun 16, 2021
8197ce3
Unified bbox coder
hhaAndroid Jun 16, 2021
af493a2
add offset
jshilong Jun 17, 2021
6040da8
remove point_bbox_coder.py
hhaAndroid Jun 17, 2021
42746c2
fix docstr
jshilong Jun 17, 2021
48dce47
merge PR5347
hhaAndroid Jun 18, 2021
17a6cca
merge PR5349
hhaAndroid Jun 18, 2021
211299a
new interface of single_proir
jshilong Jun 18, 2021
d0386ea
Merge commit 'refs/pull/5349/head' of https://github.com/open-mmlab/m…
hhaAndroid Jun 18, 2021
0927d88
fix device
jshilong Jun 18, 2021
5d2f464
Merge commit 'refs/pull/5349/head' of https://github.com/open-mmlab/m…
hhaAndroid Jun 18, 2021
d75e072
add unitest
jshilong Jun 18, 2021
946c23d
add cuda unitest
jshilong Jun 18, 2021
2ad1337
add more cuda unintest
jshilong Jun 18, 2021
5da6a00
Merge commit 'refs/pull/5349/head' of https://github.com/open-mmlab/m…
hhaAndroid Jun 18, 2021
6a5eee8
fix reppoints
jshilong Jun 18, 2021
05f18ae
Merge commit 'refs/pull/5349/head' of https://github.com/open-mmlab/m…
hhaAndroid Jun 18, 2021
19a727b
fix device
jshilong Jun 18, 2021
ed2477e
Merge commit 'refs/pull/5349/head' of https://github.com/open-mmlab/m…
hhaAndroid Jun 18, 2021
f432b9c
update all prior
hhaAndroid Jun 18, 2021
40d214c
Merge branch 'master' of https://github.com/open-mmlab/mmdetection in…
hhaAndroid Jun 18, 2021
73de85f
update vfnet
hhaAndroid Jun 18, 2021
00cf8de
add unintest for ssd and yolo and rename prior_idxs
jshilong Jun 20, 2021
e4d1c3e
add docstr for MlvlPointGenerator
jshilong Jun 21, 2021
5322cdc
Merge commit 'refs/pull/5349/head' of https://github.com/open-mmlab/m…
hhaAndroid Jun 21, 2021
6a3bf1f
update reppoints and rpnhead
hhaAndroid Jun 21, 2021
3717743
add space
jshilong Jun 22, 2021
ea79bf3
add num_base_priors
jshilong Jun 22, 2021
d02dc7f
update some model
hhaAndroid Jun 23, 2021
9299b52
update docstr
hhaAndroid Jun 23, 2021
b9fd000
fixAugFPN test and lint.
hhaAndroid Jun 23, 2021
dad4896
Merge commit 'refs/pull/5349/head' of https://github.com/open-mmlab/m…
hhaAndroid Jun 24, 2021
ce8bf29
merge master
hhaAndroid Jun 24, 2021
264dae3
Fix autoassign
hhaAndroid Jun 24, 2021
98cba8c
add docs
hhaAndroid Jun 25, 2021
a08c732
merge master
hhaAndroid Jun 25, 2021
5f55ad6
Merge branch 'master' of https://github.com/open-mmlab/mmdetection in…
hhaAndroid Jul 8, 2021
da0d9d5
Unified fcos decoding
hhaAndroid Jul 8, 2021
36b43d4
Merge branch 'master' of https://github.com/open-mmlab/mmdetection in…
hhaAndroid Jul 8, 2021
cfb6500
merge 5374
hhaAndroid Jul 8, 2021
226f833
update docstr
hhaAndroid Jul 8, 2021
f40d903
fix train error
hhaAndroid Jul 8, 2021
b04730d
Fix Vfnet
hhaAndroid Jul 12, 2021
556d5a3
Fix conflicts
hhaAndroid Jul 12, 2021
9aed4b9
Fix some
hhaAndroid Jul 12, 2021
59807b6
update centernet
hhaAndroid Jul 13, 2021
15390e2
revert
hhaAndroid Jul 13, 2021
73bb7c5
add warnings
hhaAndroid Jul 13, 2021
b9546b2
fix unittest error
hhaAndroid Jul 13, 2021
8dd5274
delete duplicated
hhaAndroid Jul 14, 2021
e09f4a8
fix comment
hhaAndroid Jul 15, 2021
f5f9486
fix docs
hhaAndroid Jul 22, 2021
f05d7df
fix type
hhaAndroid Aug 10, 2021
7fae84d
merge refactor_dense
hhaAndroid Aug 11, 2021
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
Prev Previous commit
Next Next commit
fix stride_h
  • Loading branch information
jshilong committed Jun 16, 2021
commit 07e182ef76f617c209fd461c6e240644349abce6
13 changes: 7 additions & 6 deletions mmdet/core/anchor/anchor_generator.py
Original file line number Diff line number Diff line change
Expand Up @@ -375,13 +375,14 @@ def sparse_priors(self,
"""

height, width = featmap_size
num_anchors = self.num_base_anchors[level_idx]
anchor_id = prior_indexs % num_anchors
x = (prior_indexs // num_anchors) % width
y = (prior_indexs // width // num_anchors) % height
num_base_anchors = self.num_base_anchors[level_idx]
base_anchor_id = prior_indexs % num_base_anchors
x = (prior_indexs //
num_base_anchors) % width * self.strides[level_idx][0]
y = (prior_indexs // width //
num_base_anchors) % height * self.strides[level_idx][1]
priors = torch.stack([x, y, x, y], 1).to(
dtype) * self.strides[level_idx][0] + self.base_anchors[level_idx][
anchor_id, :].to(device)
dtype) + self.base_anchors[level_idx][base_anchor_id, :].to(device)

return priors

Expand Down