読者です 読者をやめる 読者になる 読者になる

まだそんなことやってんの?

技術系備忘録ブログがメイン

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)