目标检测识别任务的训练评估结果复现不成功。能够成功复现obbdetection仓库oriented_rcnn的训练评估结果,但vitae-rvsa_dota10_ms的训练评估效果不理想
如题,根据vitae在目标检测识别上的构建方式,需要先搭建部署obbdetection的环境。我先搭好了obbdetection,准备dotav1.0的数据集,运行oriented_rcnn的faster_rcnn_orpn_r50_ms_rr_dota10.py进行训练以及评估,得到下图的效果
模型在dotav1.0上进行目标检测效果不错,说明obbdetection环境安装没问题。
随后我将remote-sensing-rvsa中object detection中的文件移动到obbdetection目录的相应位置,根据本项目readme中的命令行尝试运行faster_rcnn_orpn_our_rsp_vitae-nc-base-win-rvsa_v3_wsz7_fpn_1x_dota10_ms.py时,在backbone/init.py函数会报关于找不到"Swin"和另外几个模型的错误,我将这部分模型在__init__.py中删除后,可以正常训练评估,但效果如下图,
复现不出论文中效果。
1)先安装obbdetection,再进行本项目的代码文件和模型的移动复制,这部分流程应该是没有错误的。而后对于backbone/init.py里缺少的模型文件应该如何处理 2)项目时间也比较久了,复现过程在readme中能否详细一些。 3)这个复现结果不正确,可能是有什么原因造成的呢
@wyk201722 首先,论文中的结果都是在DOTA官方网站在线评测的,不是本地评测的,我不知道你怎么测出的86,因为目前测试集最高的是82,其次,论文中评测是将训练集和验证集混合起来一起训,然后将预测出的测试集结果打包交到评测网站上
在进行多尺度预测时,需要将训练集和测试集混合后的trnval集和测试集都进行多尺度裁剪,目前我知道有人复现了单尺度的
https://github.com/ViTAE-Transformer/Remote-Sensing-RVSA/issues/19
@wyk201722 RVSA仓库是和RSP仓库在同一套框架上跑的,RSP仓库中包含swin,只不过由于RSP已经上传,在RVSA仓库,我把其中和RVSA相关的抽了出来,应该是init.py里边还留着
@wyk201722 此外,我觉得你还可以参考一下我们上传的log文件,看看模型权重加载的对不对:https://github.com/ViTAE-Transformer/Remote-Sensing-RVSA/blob/main/Object%20Detection/logs/faster_rcnn_orpn_our_rsp_vitae-nc-base-win-rvsa_v3_wsz7_fpn_1x_dota10_ms_lr1e-4_ldr75_dpr15.log