-
Notifications
You must be signed in to change notification settings - Fork 0
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
h2rbox v2 test results? #3
Comments
yes, you are right. |
I fix a bug in h2rbox-v2 config: The previous version still used rbox supervised learning. |
Hi, I applied the changes on the h2rbox-v2 config file and I got similar results as before. In fact, I am quite confused by the def nested_projection(self, pred, target):
target_xy1 = target[..., 0:2] - target[..., 2:4] / 2
target_xy2 = target[..., 0:2] + target[..., 2:4] / 2
target_projected = torch.cat((target_xy1, target_xy2), -1)
pred_xy = pred[..., 0:2]
pred_wh = pred[..., 2:4]
da = pred[..., 4] - target[..., 4]
cosa = torch.cos(da).abs()
sina = torch.sin(da).abs()
pred_wh = torch.matmul(
torch.stack((cosa, sina, sina, cosa), -1).view(*cosa.shape, 2, 2),
pred_wh[..., None])[..., 0]
pred_xy1 = pred_xy - pred_wh / 2
pred_xy2 = pred_xy + pred_wh / 2
pred_projected = torch.cat((pred_xy1, pred_xy2), -1)
return pred_projected, target_projected
return pred_projected, target_projected Here, the target angle (i.e. the angle ground truth information) is being used to extract the |
The ground truth angle is usually 0 or pi/2 for horizontal boxes, but when the random rotation augmentation is applied, these target boxes could have an angle. "da" is the angle difference between the internal rotated box and the external circumscribed box (either rotated or horizontal). See Figure 4(b) in the paper, where the ground truth box is also a rotated box due to random rotation augmentation. The result you provided (82.8% on trainval set) seems correct. You can also run tools/test.py and upload the zip file to the DOTA server to obtain the results on test set. |
I see, so if I understood correctly: after applying random rotation the originally horizontal bounding box becomes oriented, and the code extracts the prediction's circumscribed bounding box oriented by the same angle. For the results, I would love to test them on the server but I can't get the confirmation email after sign up. It is more clear now, thanks! |
Gmail email addresses may not be able to be registered as they require the email suffix of the organization, e.g. sjtu.edu.cn, pjlab.org.cn |
Hello,
Thanks for the code, I am able to train a H2RBoxv2 model on DOTAv1 and I obtain the following results by the end of the training::
I assume these correspond to the validation scores at the end of the training, since the test annotations are not provided by the authors of DOTA. Hence, do the tables reported in the paper correspond to the results on the test set? Do you extract them by uploading the predictions on the evaluation server?
Thanks
The text was updated successfully, but these errors were encountered: