search

Thinklab-SJTU / R3Det_Tensorflow

Code for AAAI 2021 paper: R3Det: Refined Single-Stage Detector with Feature Refinement for Rotating Object
Apache License 2.0
540 stars 122 forks source link

使用DOTA_DEVKIT生成数据量的问题. #130

Closed igo312 closed 3 years ago

igo312 commented 3 years ago

作者你好，我根据论文中的要求设置按如下代码分割

split = splitbase(r'/algo/algo/renshengyong/google/dota/train',
                      r'/algo/algo/renshengyong/dataset/dota/train',
                      gap=150,
                      subsize=600,
                      num_process=8
                      )
split.splitdata(1)

但是这样我生成的图片数量有40000多张，远多于论文中提到27000，是我的ratio设置不对么？

yangxue0827 commented 3 years ago

我用的是我写的分割脚本而不是官方的

yangxue0827 commented 3 years ago

data/io/DOTA/data_crop.py

igo312 commented 3 years ago

似乎区别在于保留bbox标准上的不同。 dota要求一个裁剪至少有一个目标的IOU仍大于0.7则保留这个patch @yangxue0827 抱歉我没太理解你的做法，主要在于以下这个条件

cond2 = np.intersect1d(np.where(center_y[:] <= (bottom_right_row - top_left_row))[0] 
                                      np.where(center_x[:] <= (bottom_right_col - top_left_col))[0])

似乎是在用目标的宽高和中心点作比较，这个是什么意思呢？