ReDet icon indicating copy to clipboard operation
ReDet copied to clipboard

请问一下把这个模型用在自己的数据集上,有没有详细一点的说明

Open luoz66 opened this issue 3 years ago • 1 comments

luoz66 avatar Sep 21 '21 04:09 luoz66

你可以先尝试下载DOTA或HRSC2016,观察其格式,并将你的数据转换成类似于DOTA的格式。参考:https://github.com/csuhan/ReDet/blob/master/GETTING_STARTED.md

然后修改数据集中的类别: https://github.com/csuhan/ReDet/blob/0b9addf3c2734659fd6ffc7824f2e659fde4419c/mmdet/datasets/DOTA.py#L4-L14

接着修改对应config中的类别: https://github.com/csuhan/ReDet/blob/0b9addf3c2734659fd6ffc7824f2e659fde4419c/configs/ReDet/ReDet_re50_refpn_1x_dota1.py#L34-L59

csuhan avatar Sep 21 '21 15:09 csuhan