mmocr icon indicating copy to clipboard operation
mmocr copied to clipboard

[Feature] How to draw the epoch loss using log ? What's the meaing of loss、 loss_prob、 loss_thr、 loss_db:

Open aibohang opened this issue 1 year ago • 1 comments

What is the feature?

image

Any other context?

No response

aibohang avatar Jun 30 '23 07:06 aibohang

MMOCR hasn't officially provided this script yet, but you can implement one following MMPretrain's implementation: https://github.com/open-mmlab/mmpretrain/blob/main/tools/analysis_tools/analyze_logs.py

loss is the summation of all the sub-losses. The sub-losses' names come from DBNet's paper, and their actual implementation can be found in https://github.com/open-mmlab/mmocr/blob/1dcd6fa6958de22bcb997319833f0ac19c180ec7/mmocr/models/textdet/module_losses/db_module_loss.py

gaotongxiao avatar Jul 17 '23 07:07 gaotongxiao