DBNet.pytorch
DBNet.pytorch copied to clipboard
A pytorch re-implementation of Real-time Scene Text Detection with Differentiable Binarization
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...
真的急需,如果可以请发送到 QQ 邮箱 [email protected]
遇到训练能跑几十个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,请教下路过的大佬们