fix: image alpha error
This commit is contained in:
@@ -35,6 +35,7 @@ class OCRService:
|
||||
self.layout_detector = layout_detector
|
||||
self.image_processor = image_processor
|
||||
self.converter = converter
|
||||
|
||||
def _get_pipeline(self):
|
||||
"""Get or create PaddleOCR-VL pipeline.
|
||||
|
||||
@@ -127,15 +128,3 @@ class OCRService:
|
||||
return self.recognize_mixed(image)
|
||||
else:
|
||||
return self.recognize_formula(image)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
import cv2
|
||||
from app.services.image_processor import ImageProcessor
|
||||
from app.services.layout_detector import LayoutDetector
|
||||
image_processor = ImageProcessor(padding_ratio=0.15)
|
||||
layout_detector = LayoutDetector()
|
||||
ocr_service = OCRService(image_processor=image_processor, layout_detector=layout_detector)
|
||||
image = cv2.imread("test/image.png")
|
||||
ocr_result = ocr_service.recognize(image)
|
||||
print(ocr_result)
|
||||
Reference in New Issue
Block a user