imgsread.py

#!/usr/bin/python
# -*- coding: utf-8 -*-

import numpy as np
import cv2, os, time

# 複数の画像を読み込む関数
def imgsread(path):
    # ディレクトリ中のファイル名のリストを取得
    filelists = os.listdir(path)
    print 'number of files: ', len(filelists)
    
    # 読み込んだ画像をリストに追加
    imgs = []
    for f in filelists:
        img = cv2.imread(f)
        imgs.append(img)
        
    # リストをnumpyに変換
    return np.asarray(imgs)


# 画像の読み込み時間の計測
s1 = time.time()
array1 = imgsread('hoge/hoge/foo/bar')
print 'imgsread()', time.time() - s1

# 画像をnpy形式で保存
np.save('array.npy', array)

# npyファイルの読み込み時間の計測
s2 = time.time()
array2 = np.load('array.npy')
print 'np.load()', time.time() - s2