How to set the PRIOR_HEIGHTS and PRIOR_WEIGHTS

ZongxuPan / DrBox-v2-tensorflow

The tensorflow implementation of DrBox-v2 which is an improved detector with rotatable boxes for target detection in remote sensing images

Apache License 2.0

47 stars 21 forks source link

How to set the PRIOR_HEIGHTS and PRIOR_WEIGHTS #8

Open ChengF-Lab opened 5 years ago

ChengF-Lab commented 5 years ago

I don not know how to set the specific values of these two arrays according to my own dataset. I set a them roughly but I get runtime error around here encoded[2] = math.log(ground[2] / prior[2])

Anquanzhi commented 5 years ago

The two parameters PRIOR_HEIGHTS and PRIOR_WEIGHTS appear in pairs. In other words, the two lists should have the same length. Reasonable setting of these two parameters needs to be considered in combination with your own data. Generally, several distribution centers of the target length and width in the data set are selected as the basis for setting these two parameters. Reference: our paper "DRBox-v2:An Improved Detector with Rotatable Boxes for Target Detection in SAR Images" IV, B, 2)