open-mmlab / mmrotate

OpenMMLab Rotated Object Detection Toolbox and Benchmark
https://mmrotate.readthedocs.io/en/latest/
Apache License 2.0
1.84k stars 542 forks source link

Difficulties detecting long-thin objects #384

Open TekayaNidham opened 2 years ago

TekayaNidham commented 2 years ago

Thank you so much for your contributions.

I'm trying to train model to detect architectural objects and none of the models i train seems to detect long and thin object (as illustrated in the attached image).

and none of these exceeded 0.4 mAP for that specific class.

Do you have any architecture recommendation or a change in the config i can do to improve my results? Screenshot from 2022-06-28 09-20-00

zytx121 commented 2 years ago

I guess you can add the anchor ratios from [0.5, 1.0, 2.0] to [1/7.0, 0.5, 1.0, 2.0, 7.0]. @yangxue0827 Do you have any good suggestion?

TekayaNidham commented 2 years ago

changing it to those specific ratios didn't help can you tell me how those ratios works so i can maybe try multiple combinations

zytx121 commented 2 years ago

We will generate different ratios' anchors according to it. And you could refer to https://github.com/open-mmlab/mmdetection/blob/56e42e72cdf516bebb676e586f408b98f854d84c/mmdet/core/anchor/anchor_generator.py#L151