Detection Training - RuntimeWarning divide by zero

[ArsalanYounus007]

Hello,

I am trying to train db_resnet50 for Checkbox (detection). I created my own annotations and started the training.

I am getting
F:\GitHub\doctr\doctr\models\detection\differentiable_binarization\base.py:247: RuntimeWarning: divide by zero encountered in true_divide

Loss goes up whenever this occurs, I am sure that this is a problem with my dataset annotations, But I am unable to figure out what.

[ArsalanYounus007]

Is it because the GTs are too close to each other?? I have used the show-sample flag to get this image

[felixT2K]

Hi @ArsalanYounus007 👋🏼,

The reason why this happens is, that we filter out to small boxes which can't be a character like noise in an image which is detected as character (possible dot).
One option would be to play a bit around with the following both values to expand the seq map detections:

doctr/doctr/models/detection/differentiable_binarization/base.py

Line 35 in 7ab0ece

box_thresh: float = 0.1,

[ArsalanYounus007]

Hi, Thank you for your reply.
I have changed the value to very small for example: 0.001 - 0.01 but I still see the warning and loss jumps on each sample.

[ArsalanYounus007]

One more thing, Even though I let it train with Warnings. I got a checkpoint - I know it might not be a great checkpoint but I thought to load and test it anyways

When I load it (using: https://mindee.github.io/doctr/using_doctr/custom_models_training.html) , I get this error.

[felixT2K]

See: https://mindee.github.io/doctr/using_doctr/custom_models_training.html :)

[ArsalanYounus007]

Eish... My bad!!

[ArsalanYounus007]

As expected, Got nothing on inference
Back to my original question. How do I fix the training of Checkboxes?

[felixT2K]

Hi @ArsalanYounus007
You can try to modify the shrink value and min_size_box by lowering both values

doctr/doctr/models/detection/differentiable_binarization/base.py

Line 165 in 7ab0ece

shrink_ratio = 0.4

or expand your box coordinates in your dataset a bit.
How many samples are in your dataset ?

[ArsalanYounus007]

I have around 1500 pages in total with CheckBoxes.
I will play with these parameters and see if it works, Thank you!!