search

jbwang1997 / OBBDetection

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

Samples_per_gpu set to 1 when inference #25

Closed YangHai-1218 closed 2 years ago

YangHai-1218 commented 2 years ago

Hi, thanks for your work! But I was just wondering why you have to set the samples_per_gpu=1 when inference. Does there exist some bug? Or some other concern is within it?

jbwang1997 commented 2 years ago

It's a legacy problem from MMDetection earlier vision. This repo is based on the MMdetection v2.2 where all models' post-processing are written under the condition that batch size is 1.

In higher vision MMDetection, this problem has been fixed.

YangHai-1218 commented 2 years ago

Thanks! But do you have any plan to fix this problem or just update to a higher version MMDetection?

jbwang1997 commented 2 years ago

I have a plan to update the version of MMDetection, but it's a big project.😖 The update maybe takes a long time.

YangHai-1218 commented 2 years ago

Thanks! It is truly a big project and it is really not easy to fix the problem. Anyway, looking forward to the updated version!