jbwang1997 / OBBDetection

OBBDetection is an oriented object detection library, which is based on MMdetection.
Apache License 2.0
520 stars 111 forks source link

参数代表的含义 #108

Open a1b2c3s4d4 opened 2 years ago

a1b2c3s4d4 commented 2 years ago

/DOTA_OBBDetection/mmdet/models/roi_heads/roi_extractors/obb/obb_single_level_roi_extractor.py文件中的 def forward(self, feats, rois, roi_scale_factor=None): 这里的feats是指什么呀? 下面是输出的rois,请问每一行的六个数是什么含义呐?感觉像是中点偏移表示中的(x,y,w,h,▲alpha, ▲beta ),但是数值又对不上。。。求解答

    # tensor([[0.0000e+00, 1.8100e+02, 9.1600e+02, 2.8410e+01, 1.0993e+01,
    #          -3.1059e+00],
    #         [0.0000e+00, 7.3223e+02, 9.3639e+02, 2.1954e+01, 7.5611e+00,
    #          -1.0460e+00],
    #         [0.0000e+00, 7.9756e+02, 3.0058e+02, 2.4200e+01, 1.0800e+01,
    #          -2.4981e+00],
    #         ...,
    #         [3.0000e+00, 7.6626e+02, 8.8140e+02, 2.3675e+01, 2.3083e+01,
    #          -7.8036e-01],
    #         [3.0000e+00, 7.1368e+02, 1.1127e+01, 1.0663e+02, 9.2984e+01,
    #          -8.0230e-01],
    #         [3.0000e+00, 5.0430e+02, 3.9442e+02, 3.5547e+01, 2.8124e+01,
    #          -8.8352e-01]], device='cuda:2')
jbwang1997 commented 2 years ago

feats是图像特征,rois六个参数分别表示(batch_index, x, y, w, h, theta),其中batch_index表示这个roi所在图片在一个batch中的序号。

a1b2c3s4d4 commented 2 years ago

feats是图像特征,rois六个参数分别表示(batch_index, x, y, w, h, theta),其中batch_index表示这个roi所在图片在一个batch中的序号。

请问这个theta的取值范围是多少呀?请问这个送入obb_single_level_roi_extractor.py文件的rois是哪个文件生成的呀?谢谢