python - 获取 TypeError：'TextDetector' 对象不可调用

Question

我的 TextDetector 对象是一个类的蓝图，但仍然无法调用它。它以前可以工作，但现在无法正常工作。不知道为什么会这样。你们中的任何人都可以给我这个 TypeError 的原因吗？

导入 predict_det

#from predict_det 导入文本检测器

导入简历2

#进口火炬

#device = torch.device('cuda:0')

​</p>

text_detector = predict_det.TextDetector()

img = cv2.imread(r'C:\Users\pearl\Downloads\AI\Project OCR\OCR_PP_tester\Temp_img_folder\images.jpg')

dt_boxes, elapse = text_detector(img)

<类'列表'>

---

TypeError Traceback（最近一次调用）~\AppData\Local\Temp/ipykernel_3160/1748103168.py in 7 text_detector = predict_det.TextDetector() 8 img = cv2.imread(r'C:\Users\pearl\Downloads\AI\项目 OCR\OCR_PP_tester\Temp_img_folder\images.jpg') ----> 9 dt_boxes, elapse = text_detector(img)

TypeError：“TextDetector”对象不可调用

Answer 1

您没有为 TextDetector 构造函数传递任何参数，因此 det_algorithm 为“空”，这会引发错误：

    if self.det_algorithm == "EAST":
        preds['f_geo'] = outputs[0]
        preds['f_score'] = outputs[1]
    elif self.det_algorithm == 'SAST':
        preds['f_border'] = outputs[0]
        preds['f_score'] = outputs[1]
        preds['f_tco'] = outputs[2]
        preds['f_tvo'] = outputs[3]
    elif self.det_algorithm == 'DB':
        preds['maps'] = outputs[0]
    else:
        raise NotImplementedError