回答編集履歴

1 importミス

wakame

wakame score 1135

2018/06/20 18:39  投稿

いちばん簡単なのは自分で実装してしまうことでしょうか。(コピペしただけですけど)
importから`list_pictures`外しておいてくださいね。
```python
# coding:utf-8
import os  
import re  
import keras
import numpy as np
from keras.models import load_model
from keras.preprocessing.image import array_to_img, img_to_array, load_img
def list_pictures(directory, ext='jpg|jpeg|bmp|png|ppm'):
   return [os.path.join(root, f)
           for root, _, files in os.walk(directory) for f in files
           if re.match(r'([\w]+\.(?:' + ext + '))', f.lower())]
model = load_model('mnist_model.h5')
for picture in list_pictures(r'C:\Users\Desktop\pic'):
   X = []
   img = img_to_array(
       load_img(picture, target_size=(28, 28), grayscale=True))
   X.append(img)
   X = np.asarray(X)
   X = X.astype('float32')
   X = X / 255.0
   features = model.predict(X)
   print('----------')
   print(picture)
   print(features.argmax())
   print('----------')
```
[keras-preprocessing/keras_preprocessing/image.py](https://github.com/keras-team/keras-preprocessing/blob/master/keras_preprocessing/image.py#L494-L497)

思考するエンジニアのためのQ&Aサイト「teratail」について詳しく知る