训练自己数据到第三轮报错
不明白第一轮正常跑完,到第三报错了,看报错原因应该是数据的问题
2023-03-22 20:52:55,756 - davarocr - INFO - Epoch [3][2610/8424] lr: 1.000e-02, eta: 1 day, 11:59:42, time: 2.943, data_time: 0.423, memory: 15020, loss_rpn_cls: 0.0664, loss_rpn_bbox: 0.0797, loss_cls: 0.1295, acc: 94.2090, loss_bbox: 0.1107, loss_mask: 0.1790, loss_lpma: 0.4966, loss_seg: 0.1608, loss_reg: 0.9659, loss: 2.1887, grad_norm: 4.5623 2023-03-22 20:53:12,968 - davarocr - INFO - Epoch [3][2620/8424] lr: 1.000e-02, eta: 1 day, 11:59:12, time: 1.721, data_time: 0.185, memory: 15020, loss_rpn_cls: 0.0824, loss_rpn_bbox: 0.0613, loss_cls: 0.1165, acc: 94.7233, loss_bbox: 0.1174, loss_mask: 0.2121, loss_lpma: 0.5267, loss_seg: 0.1736, loss_reg: 0.7531, loss: 2.0431, grad_norm: 4.1360 /root/workspace/DAVAR-Lab-OCR-main/davarocr/davar_table/core/mask/lp_mask_target.py:55: RuntimeWarning: Mean of empty slice. middle_x, middle_y = round(np.where(box_text == 1)[1].mean()), round(np.where(box_text == 1)[0].mean()) /root/miniconda/envs/d2/lib/python3.7/site-packages/numpy/core/_methods.py:189: RuntimeWarning: invalid value encountered in double_scalars ret = ret.dtype.type(ret / rcount)
File "/root/workspace/DAVAR-Lab-OCR-main/davarocr/davar_table/core/mask/lp_mask_target.py", line 55, in get_lpmask_single middle_x, middle_y = round(np.where(box_text == 1)[1].mean()), round(np.where(box_text == 1)[0].mean())
请问您解决这个问题了没?
@leonodelee 没有呢!
请问解决了吗,遇到了同样的问题,在训练了一会以后就报这个错误
@tianyongliu 请问你的数据集格式是怎么样的呢?