jbwang1997 / OBBDetection

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

训练时不过滤无目标图片 #30

Closed liliwannian closed 2 years ago

liliwannian commented 2 years ago

您好,我想把所有图片放进去训练,就是有目标和没目标都进行训练,请问应该怎么修改?希望您有空能够解答,感谢!

jbwang1997 commented 2 years ago

在dataset中会将没有目标的图像块过滤,如果要训练没有目标的图像块需要修改dataset的代码

之后在pipeline里面也会有FliterEmpty去过滤没有目标的图像块,需要删除掉。

liliwannian commented 2 years ago

已成功修改,非常感谢!