DBNet.pytorch icon indicating copy to clipboard operation
DBNet.pytorch copied to clipboard

A pytorch re-implementation of Real-time Scene Text Detection with Differentiable Binarization

Results 103 DBNet.pytorch issues
Sort by recently updated
recently updated
newest added

Traceback (most recent call last): File "/content/DBNet.pytorch/tools/eval.py", line 78, in eval = EVAL(args.model_path) File "/content/DBNet.pytorch/tools/eval.py", line 31, in __init__ checkpoint = torch.load(model_path, map_location=torch.device('cpu')) File "/usr/local/lib/python3.10/site-packages/torch/serialization.py", line 815, in load return...

遇到训练能跑几十个epoch后,突然在test module时候会报在/utils/ocr_metric/icdar2015/detection/iou.py的第12行的“rotatedRectangleIntersection(rect_a, rect_b)”错误,找不到错误的头绪

找到/post_processing/seg_detector_representer.py,找到def boxes_from_bitmap(),在105行, 原本的“contours, _ = cv2.findContours((bitmap * 255).astype(np.uint8), cv2.RETR_LIST, cv2.CHAIN_APPROX_SIMPLE)”改成, “_, contours, _ = cv2.findContours((bitmap * 255).astype(np.uint8), cv2.RETR_LIST, cv2.CHAIN_APPROX_SIMPLE)”

请问有没有最后得到的结果

希望能对齐[chineseocr_lite](https://github.com/DayBreak-u/chineseocr_lite)的dbnet,请教下路过的大佬们