search

shaunyuan22 / SODA-mmrotate

SODA-A Small Object Detection Toolbox and Benchmark
https://shaunyuan22.github.io/SODA/
Apache License 2.0
37 stars 6 forks source link

[Feature] The test process takes too long. #1

Closed bozihu closed 1 year ago

bozihu commented 1 year ago

What's the feature?

When I inference the test data, the merge process and evaluation process take too long. For example, when I use Rotated Retinanet with Res50 as backbone to test, it will cost half an hour. Please give me some advises.

Any other context?

No response

shaunyuan22 commented 1 year ago

The evaluation process stuck in the IoU computation stage which could get worse when there are dense predictions and instances. We have already organized the codes in a multi-processing manner hence increase the args nproc could alleviate such situation.

Songyeyaosong commented 1 year ago

Hi, I have changed some code to solve this problem. Hope this can be helpful. shaunyuan22/SODA-mmrotate#7