pythonのopencvを用いてtensorflowでの学習に用いる画像のリサイズプログラム

tensorflowを用いてCIFAR-10のようなデータセットを作るために画像を集め、 それをあるディレクトリにまとめて保存した際そのディレクトリ内の 画像を学習用にリサイズするプログラム

# -*- coding: utf-8 -*-

import cv2
import numpy as np
import os

#file_dirにオリジナルのサイズの画像を格納しているディレクトリのpath指定
#resize_dirに出力先ディレクトリのpath指定

file_dir = 'オリジナルサイズ画像のディレクトリに適宜変更'
resize_dir = '出力先のディレクトリに適宜変更'
files = os.listdir(file_dir)
for i, file in enumerate(files):
    if i > 0:
        file_path = file_dir + '/' + file
        img = cv2.imread(file_path, cv2.IMREAD_COLOR)

        size = (28, 28)
        resize_img = cv2.resize(img, size)

        cv2.imwrite(resize_dir + '/' + file, resize_img)