SDMGRModuleLoss¶
- class mmocr.models.kie.SDMGRModuleLoss(weight_node=1.0, weight_edge=1.0, ignore_idx=- 100)[源代码]¶
The implementation the loss of key information extraction proposed in the paper: Spatial Dual-Modality Graph Reasoning for Key Information Extraction.
- 参数
- 返回类型
- forward(preds, data_samples)[源代码]¶
Forward function.
- 参数
preds (tuple(Tensor, Tensor)) –
data_samples (list[KIEDataSample]) – A list of datasamples containing
gt_instances.labels
andgt_instances.edge_labels
.
- 返回
Loss dict, containing
loss_node
,loss_edge
,acc_node
andacc_edge
.- 返回类型