Open OE-jimvanoosten opened 1 month ago
I was wondering why _withcls=True in the RotatedBBoxHead class (parent class of ORTH_RotatedConvFCBBoxHead class)?
This way you are creating a fully connected layer for classification, which is not being used as the classification branch is replaced by the cosine similarity calculation between the orthogonal vectors and the image features.
Happy to hear from you!
Thanks in advance.
Thank you for your careful observation. Indeed, the with_cls=True setting is no longer needed here, as the convolutional or fully connected layers have already been replaced.
I was wondering why _withcls=True in the RotatedBBoxHead class (parent class of ORTH_RotatedConvFCBBoxHead class)?
This way you are creating a fully connected layer for classification, which is not being used as the classification branch is replaced by the cosine similarity calculation between the orthogonal vectors and the image features.
Happy to hear from you!
Thanks in advance.