Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

目标检测识别任务的训练评估结果复现不成功。能够成功复现obbdetection仓库oriented_rcnn的训练评估结果,但vitae-rvsa_dota10_ms的训练评估效果不理想 #33

Open
wyk201722 opened this issue Apr 19, 2024 · 3 comments

Comments

@wyk201722
Copy link

如题,根据vitae在目标检测识别上的构建方式,需要先搭建部署obbdetection的环境。我先搭好了obbdetection,准备dotav1.0的数据集,运行oriented_rcnn的faster_rcnn_orpn_r50_ms_rr_dota10.py进行训练以及评估,得到下图的效果
resnetobb
模型在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中删除后,可以正常训练评估,但效果如下图,
12
复现不出论文中效果。

1)先安装obbdetection,再进行本项目的代码文件和模型的移动复制,这部分流程应该是没有错误的。而后对于backbone/init.py里缺少的模型文件应该如何处理
2)项目时间也比较久了,复现过程在readme中能否详细一些。
3)这个复现结果不正确,可能是有什么原因造成的呢

@DotWang
Copy link
Collaborator

DotWang commented Apr 19, 2024

@wyk201722 首先,论文中的结果都是在DOTA官方网站在线评测的,不是本地评测的,我不知道你怎么测出的86,因为目前测试集最高的是82,其次,论文中评测是将训练集和验证集混合起来一起训,然后将预测出的测试集结果打包交到评测网站上

在进行多尺度预测时,需要将训练集和测试集混合后的trnval集和测试集都进行多尺度裁剪,目前我知道有人复现了单尺度的

#19

@DotWang
Copy link
Collaborator

DotWang commented Apr 19, 2024

@wyk201722 RVSA仓库是和RSP仓库在同一套框架上跑的,RSP仓库中包含swin,只不过由于RSP已经上传,在RVSA仓库,我把其中和RVSA相关的抽了出来,应该是init.py里边还留着

@DotWang
Copy link
Collaborator

DotWang commented Apr 19, 2024

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants