dvlab-research / GridMask

281 stars 55 forks source link

Question about the size of mask! #12

Closed sky-fly97 closed 3 years ago

sky-fly97 commented 4 years ago

Hello, when I use the grid mask in object detection, I find that the generated default mask box is quite large. I have a question. If such a large black box covers the target, will it not affect the training? Or do I need to set the size of the black box according to the bbox size in each sample, and the size of the black box should not exceed 2 / 3 or 1 / 2 of the minimum target? Do I need to think that way?

sky-fly97 commented 4 years ago

{HGYH57(MRU8`9EO(Y~VQ{S

sky-fly97 commented 4 years ago

image image

akuxcw commented 4 years ago

Hello, the default setting is suitable for COCO dataset. This may because the original training of COCO is easy to overfit, even removing some objects hardly affects the training. If you are using other datasets, you may decrease the mask box. But in my opinion, 1 / 2 of the minimum target is too small (in COCO the smallest object‘s side length is smaller than 30 ). If you find the default setting will cause underfitting in your experiments, maybe decrease the probability of applying GridMask is a better way.

sky-fly97 commented 4 years ago

Hello, the default setting is suitable for COCO dataset. This may because the original training of COCO is easy to overfit, even removing some objects hardly affects the training. If you are using other datasets, you may decrease the mask box. But in my opinion, 1 / 2 of the minimum target is too small (in COCO the smallest object‘s side length is smaller than 30 ). If you find the default setting will cause underfitting in your experiments, maybe decrease the probability of applying GridMask is a better way.

Thanks very much!!! Another question is that if my ori train process is 18 epoch, and I use grid mask at 0.8 prob, do I need to set epoch to twice or more?

akuxcw commented 4 years ago

Usually, double the training time will increase the performance with GridMask. For your reference, in COCO the performance will increase until 4X original training time.

sky-fly97 commented 4 years ago

Usually, double the training time will increase the performance with GridMask. For your reference, in COCO the performance will increase until 4X original training time.

Thanks, I got it!

sky-fly97 commented 4 years ago

Usually, double the training time will increase the performance with GridMask. For your reference, in COCO the performance will increase until 4X original training time.

I'm sorry for the last question. Does grid mask need to be placed after image normalization, that is, at the end of all augments?

akuxcw commented 4 years ago

Yes, GridMask should be placed after normalization.