search

yijingru / BBAVectors-Oriented-Object-Detection

[WACV2021] Oriented Object Detection in Aerial Images with Box Boundary-Aware Vectors
MIT License
466 stars 87 forks source link

请问根据论文，测试结果根据alpha分为HBB和OBB两种，而我在test.py里好像没有看到您有这样分？不知道是不是我哪里理解错了忘解答，感谢！ #60

Open AIFANLIN opened 3 years ago

AIFANLIN commented 3 years ago

请问根据论文，测试结果根据alpha分为HBB和OBB两种，而我在test.py里好像没有看到您有这样分？

rexyang94 commented 3 years ago

In decoder.py，you can see the author generate a mask of area that cls_theta > 0.8 in line 70 cls_theta = cls_theta.view(batch, self.K, 1) mask = (cls_theta>0.8).float().view(batch, self.K, 1)