ruoqianguo / DetNet_pytorch

An implementation of DetNet: A Backbone network for Object Detection.
MIT License
377 stars 117 forks source link

something about 'MAX_NUM_GT_BOXES' #3

Closed baolinhu closed 6 years ago

baolinhu commented 6 years ago

I want to use your code to train your own dataset. There are many gt boxes in it, and there are probably hundreds of them.MAX_NUM_GT_BOXES=20, can I change this parameter a lot? I see the default in Pascal VOC is 20.

ruoqianguo commented 6 years ago

i think you can change that if you need (if your data gt boxes are less than 20).

baolinhu commented 6 years ago

@guoruoqian Thanks for your reply. I mean my dataset has much more objects in one image.

ruoqianguo commented 6 years ago

@baolinhu ,hi

  1. if your GT boxes is greater than 20, there will be some boxes ignored.

  2. I think if you set MAX_NUM_GT_BOXES=100. it will train slower and not effect AP.