コード例 #1
0
ファイル: ocr.py プロジェクト: gouvina/poc-movie-classifier
def get_resized_image_text(filename):
    ''' Get text from image using different sizes '''
    img = cv.imread(filename, 0)
    img_large = optimizer.enlarge_image(img)
    img_small = optimizer.reduce_image(img)
    return [
        parser.clean_text(tesseract.image_to_string(img))
        for img in [img_large, img_small]
    ]
コード例 #2
0
ファイル: ocr.py プロジェクト: gouvina/poc-movie-classifier
def get_threshold_image_text(filename, resize_type):
    ''' Get text from image using different types of threshold '''
    img = cv.imread(filename, 0)
    if resize_type == ResizeTypes.ENLARGE:
        img = optimizer.enlarge_image(img)
    elif resize_type == ResizeTypes.REDUCE:
        img = optimizer.reduce_image(img)
    images = optimizer.get_image_thresholds(img)
    return [
        parser.clean_text(tesseract.image_to_string(img)) for img in images
    ]
コード例 #3
0
ファイル: ocr.py プロジェクト: gouvina/poc-movie-classifier
def get_image_text(filename):
    ''' Get text from image using tesseract '''
    img = cv.imread(filename, 0)
    return parser.clean_text(tesseract.image_to_string(img))
コード例 #4
0
ファイル: ocr.py プロジェクト: gouvina/poc-movie-classifier
def get_reduced_image_text(filename):
    ''' Get text from image with half size '''
    img = cv.imread(filename, 0)
    img_small = optimizer.enlarge_image(img)
    return parser.clean_text(tesseract.image_to_string(img_small))
コード例 #5
0
ファイル: ocr.py プロジェクト: gouvina/poc-movie-classifier
def get_enlarged_image_text(filename):
    ''' Get text from image with double size '''
    img = cv.imread(filename, 0)
    img_large = optimizer.enlarge_image(img)
    return parser.clean_text(tesseract.image_to_string(img_large))