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

Question

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

导入 predict_det

#from predict_det import TextDetector

导入 cv2

#import 火炬

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

​

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 在 7 text_detector = predict_det.TextDetector() 8 img = cv2.imread(r'C:\Users\pearl\Downloads\AI\Project 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