训练HRSC数据集
Opened this issue · 6 comments
您好,想训练以下HRSC数据集,在训练之前需要对HRSC数据集做哪些处理呢,这里没用弄明白,期待回复
@mifuheng 你好,刚上传了HRSC2DOTA.py脚本。 首先,用HRSC2DOTA.py将hrsc数据集的原始标注转化dota的标注形式(txt), 再通过DOTA2COCO_poly.py 转化为coco格式的json annotation形式,注意生成的train/val.json中的bbox是poly(8个值)形式。
您好,谢谢回复,我现在已经将HRSC数据集利用给出的dota配置文件训练了一下,请问如何对结果进行评估呢,通过test代码生成了pkl文件,但如何对pkl文件进行评估还是不明白,想请教一下
@mifuheng 对pkl文件进行解析,解析代码参考parse_pkl_for_evaluation.py 生成txt的结果,再用hrsc2016_evaluation.py进行评估。 Sorry to rely late.
您好,谢谢回复,我现在已经将HRSC数据集利用给出的dota配置文件训练了一下,请问如何对结果进行评估呢,通过test代码生成了pkl文件,但如何对pkl文件进行评估还是不明白,想请教一下。
您好,感谢您能提出这个问题,并做出进一步的回答。我在使用HRSC2016数据集也遇到了同样的问题,不知道您最后怎么实现的,期待您的回复,祝您科研顺利。
您好,谢谢回复,我现在已经将HRSC数据集利用给出的dota配置文件训练了一下,请问如何对结果进行评估呢,通过test代码生成了pkl文件,但如何对pkl文件进行评估还是不明白,想请教一下
这个pkl文件我用test.py和tools/test.py都报bug ,请问您是怎么做的