ConsistentTeacher
ConsistentTeacher copied to clipboard
AnchorGenerator can support multiply scales for ratios?
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?
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.