jbwang1997 / OBBDetection

OBBDetection is an oriented object detection library, which is based on MMdetection.
Apache License 2.0
522 stars 112 forks source link

使用obb中的fcos训练HRSC数据集,召回率和AP都为0,请问是为什么啊? #92

Open Ro-xjy opened 2 years ago

Ro-xjy commented 2 years ago

1639448230(1)

jbwang1997 commented 2 years ago

希望能提供额外信息。

Ro-xjy commented 2 years ago

您好,大佬,我是把您的OBBDetection框架下载下来mmdetection 2.2.0,跑的这里面的fcos_obb_r50_caffe_fpn_gn-head_4x4_1x_dota10.py 训练命令: bash ./tools/dist_train.sh configs/obb/fcos_obb/fcos_obb_r50_caffe_fpn_gn-head_4x4_1x_dota10.py 2 测试结果就是召回率和AP都为0,请问您跑这个fcos_obb有问题吗? 我用的数据集是HRSC数据集,修改num_classes=1,_base_改为'../base/datasets/hrsc.py',train_pipeline和test_pipeline也改为hrsc.py中的,我用这个数据集跑了您的框架里的其他的模型,没问题。 恳请您提供一些建议,谢谢大佬。

Ro-xjy commented 2 years ago

还有就是caffe模型没有办法下载下来,我使用的是pytorch的预训练权重

jbwang1997 commented 2 years ago

HRSC数据集图像数量较少,需要多训练几个epoch

Ro-xjy commented 2 years ago

HRSC数据集图像数量较少,需要多训练几个epoch

好的,我试试,谢谢您,辛苦啦

ghost commented 2 years ago

您好,请问您的问题解决了吗,我也尝试在focs上训练hrsc数据集,改动与您类似,并将训练轮数改为了36轮,但结果还是0

ghost commented 2 years ago

您好,我修改之后的hrsc配置文件可以正常训练了,初步结果是map为0.71,你可以试一下。链接: https://pan.baidu.com/s/1SdWMNIxknkAqsByhp22xeg 密码: moag

LowPassFilter-phil commented 2 years ago

我跑出的dota1.0也全部都是0,你知道这是怎么回事吗?配置文件他提供的就是dota的。 Start calculate mAP!!! Result is Only for reference, final result is subject to DOTA_devkit

+--------------------+-----+--------+--------+--------+ | class | gts | dets | recall | ap | +--------------------+-----+--------+--------+--------+ | large-vehicle | 0 | 100836 | 0.0000 | 0.0000 | | swimming-pool | 0 | 11648 | 0.0000 | 0.0000 | | helicopter | 0 | 8037 | 0.0000 | 0.0000 | | bridge | 0 | 43041 | 0.0000 | 0.0000 | | plane | 0 | 21394 | 0.0000 | 0.0000 | | ship | 0 | 69761 | 0.0000 | 0.0000 | | soccer-ball-field | 0 | 8383 | 0.0000 | 0.0000 | | basketball-court | 0 | 5721 | 0.0000 | 0.0000 | | ground-track-field | 0 | 9289 | 0.0000 | 0.0000 | | small-vehicle | 0 | 100046 | 0.0000 | 0.0000 | | baseball-diamond | 0 | 8011 | 0.0000 | 0.0000 | | tennis-court | 0 | 11894 | 0.0000 | 0.0000 | | roundabout | 0 | 8534 | 0.0000 | 0.0000 | | storage-tank | 0 | 31239 | 0.0000 | 0.0000 | | harbor | 0 | 40260 | 0.0000 | 0.0000 | +--------------------+-----+--------+--------+--------+ | mAP | | | | 0.0000 | +--------------------+-----+--------+--------+--------+

LowPassFilter-phil commented 2 years ago

你在voc里面的训练类别是怎么写的,就写ship就行了吗

ghost commented 2 years ago

你好,我的配置文件就是参照其余检测HRSC数据集模型的配置文件写的,没有涉及到voc

------------------ 原始邮件 ------------------ 发件人: @.>; 发送时间: 2022年3月10日(星期四) 上午9:29 收件人: @.>; 抄送: @.>; @.>; 主题: Re: [jbwang1997/OBBDetection] 使用obb中的fcos训练HRSC数据集,召回率和AP都为0,请问是为什么啊? (Issue #92)

你在voc里面的训练类别是怎么写的,就写ship就行了吗

— Reply to this email directly, view it on GitHub, or unsubscribe. Triage notifications on the go with GitHub Mobile for iOS or Android. You are receiving this because you commented.Message ID: @.***>

Panhonganadd commented 3 months ago

我跑data1.5验证的结果也全是0 image

Panhonganadd commented 3 months ago

您好,我修改之后的hrsc配置文件可以正常训练了,初步结果是map为0.71,你可以试一下。链接: https://pan.baidu.com/s/1SdWMNIxknkAqsByhp22xeg 密码: moag 你好,请问你怎么修改的,能不能再发一次