ReDet
ReDet copied to clipboard
请问一下把这个模型用在自己的数据集上,有没有详细一点的说明
你可以先尝试下载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