search

open-mmlab / mmocr

OpenMMLab Text Detection, Recognition and Understanding Toolbox
https://mmocr.readthedocs.io/en/dev-1.x/
Apache License 2.0
4.21k stars 738 forks source link

Ask about loss of ABCNet #1930

Open srcjun opened 1 year ago

srcjun commented 1 year ago

Branch

1.x branch https://mmocr.readthedocs.io/en/dev-1.x/

📚 The doc issue

Hello, I just trained ABCNet and the pasted picture is my ABCNet training log. I want to ask two things about the loss which is highlighted with red box in the picture.

  1. What does the loss mean? It is summation of text detection loss and text recognition loss?
  2. Where can I find the file(ex. py file) related to the loss? This are questions about the loss. Thanks. loss

Suggest a potential alternative/fix

No response

gaotongxiao commented 1 year ago
  1. Right
  2. https://github.com/open-mmlab/mmocr/blob/main/projects/ABCNet/abcnet/model/abcnet_det_module_loss.py
srcjun commented 1 year ago

Thank you for reply! but my second question was about the total loss(Detection loss + Recognition loss). I think there are only loss_cls, loss_bbox, loss_centerness, loss_bezier in abcenet_det_module_loss.py. I want to know where the total loss is calculated. Thanks.