Closed geobao closed 2 years ago
the num_classes
in bbox_head should be 1 when using hrsc dataset.
This was it, thanks! I re-trained the model with the new config. The testing gives me this output:
+-------+------+-------+--------+--------+
| class | gts | dets | recall | ap |
+-------+------+-------+--------+--------+
| ship | 1188 | 44519 | 0.0000 | 0.0000 |
+-------+------+-------+--------+--------+
| mAP | | | | 0.0000 |
+-------+------+-------+--------+--------+
I used 36 epochs. I will try with more epochs, but I am not convinced it is that because the scores are not low but exactly 0.0000. I will let you know the result with more epochs
lr_config = dict( policy='step', warmup='constant', warmup_iters=500, warmup_ratio=1.0 / 3, step=[8, 11]) total_epochs = 36
It seems you only extend the total epochs to 36. The first and second steps to decay the lr should also be extended. 8 -> 24 11 -> 33
thanks, I will try that and let you know how it goes
This was the result. It makes more sense now. Thanks!
+-------+------+------+--------+--------+
| class | gts | dets | recall | ap |
+-------+------+------+--------+--------+
| ship | 1188 | 9589 | 0.7727 | 0.6331 |
+-------+------+------+--------+--------+
| mAP | | | | 0.6331 |
+-------+------+------+--------+--------+
I would like to ask you. Is this implementation of FCOS the rotated version as in:
FCOSR: A Simple Anchor-free Rotated Detector for Aerial Object Detection https://arxiv.org/pdf/2111.10780.pdf
or is it rather the original FCOS for generic purpose (rotated or horizontal labeled objects).
Thanks
Answering my last question: No, it is not FCOSR. This is the original FCOS with the loss "PolyIoULoss" for obb
I am trying to run OBB with FCOS in dataset HRSC2016. To do so I have based the config.py on an example of OBB with FCOS in dataset DOTA and removed these two parts:
dict(type='LoadDOTASpecialInfo')
dict(type='DOTASpecialIgnore', ignore_size=2)
I managed to execute the training with no problem. Then, when I try to run the testing, I get this error. What am I doing wrong?
The configuration file is: