search

Adamdad / ConsistentTeacher

[CVPR2023 Highlight] Consistent-Teacher: Towards Reducing Inconsistent Pseudo-targets in Semi-supervised Object Detection
Apache License 2.0
291 stars 19 forks source link

AnchorGenerator can support multiply scales for ratios? #28

Open zimenglan-sysu-512 opened 1 year ago

zimenglan-sysu-512 commented 1 year ago

when set multiply ratios like [0.5, 1, 2.0], it will crash in the FAM3DHead.

  File "/home/tiger/ConsistentTeacher/ssod/models/dense_heads/fam3d.py", line 169, in forward
    b, h, w, 4).permute(0, 3, 1, 2) / stride[0]
RuntimeError: shape '[4, 104, 180, 4]' is invalid for input of size 898560

so will support multiply scales for ratios?

Johnson-Wang commented 1 year ago

Sorry. It does not work for multiple ratios at present. But you may try to change the shape size in this line of code to support this feature.