1. 程式人生 > >OpenCV修改圖片大小

OpenCV修改圖片大小

因為最近想實現驗證碼和手寫漢字的識別,在採集處理資料的時候經常要處理大小不同的圖片,沒有找到一個滿意的程式,後面決定自己動手豐衣足食。使用了一下OpenCV庫實現程式很簡單,程式碼就是註釋。話不多說,直接上程式碼。

# -*- coding=utf-8 -*-
import os
import sys
import cv2

#輸入圖片的
input_dir = './input_img'
output_dir = './output_img'

width = 160
height = 60

if not os.path.exists(output_dir):
    os.makedirs(output_dir)

index = 1
for (path,dirnames,filenames) in os.walk(input_dir):
    for filename in filenames:
        if filename.endswith('.jpg'):
            print('正在處理第 %s 張圖片' % index)
            img_path = path + '/' + filename
            img = cv2.imread(img_path)
            new_img = cv2.resize(img,(width,height))
            cv2.imwrite(output_dir + '/' + str(index) + '.jpg', new_img)
            index += 1
            key = cv2.waitKey(30) & 0xff
            if key == 27:
                sys.exit(0)


使用過後,發現處理圖片的速度也是非常快的