LiWentomng/OrientedRepPoints

训练HRSC数据集

Opened this issue · 6 comments

mifuheng commented
训练HRSC数据集
mifuheng commented

您好，想训练以下HRSC数据集，在训练之前需要对HRSC数据集做哪些处理呢，这里没用弄明白，期待回复

LiWentomng commented

@mifuheng 你好，刚上传了HRSC2DOTA.py脚本。 首先，用HRSC2DOTA.py将hrsc数据集的原始标注转化dota的标注形式(txt), 再通过DOTA2COCO_poly.py 转化为coco格式的json annotation形式，注意生成的train/val.json中的bbox是poly（8个值）形式。

mifuheng commented

您好，谢谢回复，我现在已经将HRSC数据集利用给出的dota配置文件训练了一下，请问如何对结果进行评估呢，通过test代码生成了pkl文件，但如何对pkl文件进行评估还是不明白，想请教一下

LiWentomng commented

@mifuheng 对pkl文件进行解析，解析代码参考parse_pkl_for_evaluation.py 生成txt的结果，再用hrsc2016_evaluation.py进行评估。 Sorry to rely late.

zzz996259 commented

您好，谢谢回复，我现在已经将HRSC数据集利用给出的dota配置文件训练了一下，请问如何对结果进行评估呢，通过test代码生成了pkl文件，但如何对pkl文件进行评估还是不明白，想请教一下。
您好，感谢您能提出这个问题，并做出进一步的回答。我在使用HRSC2016数据集也遇到了同样的问题，不知道您最后怎么实现的，期待您的回复，祝您科研顺利。

H1d30nbu3h commented

您好，谢谢回复，我现在已经将HRSC数据集利用给出的dota配置文件训练了一下，请问如何对结果进行评估呢，通过test代码生成了pkl文件，但如何对pkl文件进行评估还是不明白，想请教一下

这个pkl文件我用test.py和tools/test.py都报bug ，请问您是怎么做的