In [1]:

    

import glob
import numpy as np
import csv
from keras.models import load_model
from keras.preprocessing import image
classifier = load_model('/demo5/all-model.h5')
with open('/demo5/prediction.csv', "w") as csv_file:
    writer = csv.writer(csv_file, delimiter=',')
    for filename in glob.iglob('/demo5/Remaining Clips/*.jpg'):
        test_image = image.load_img( filename,target_size=(64, 64))
        test_image = image.img_to_array(test_image)
        test_image = np.expand_dims(test_image, axis = 0)
        result = classifier.predict_classes(test_image, verbose = 1)
        for item in train_generator.class_indices:   # Python's for loops are a "for each" loop             
            if (result[0] == train_generator.class_indices[item]):
                print(filename, item)
                line = filename + ',' + item
                writer.writerow([filename, item])


    

    




Using TensorFlow backend.






    




1/1 [==============================] - 0s






    




---------------------------------------------------------------------------
NameError                                 Traceback (most recent call last)
<ipython-input-1-3f67342c9ed2> in <module>()
     12         test_image = np.expand_dims(test_image, axis = 0)
     13         result = classifier.predict_classes(test_image, verbose = 1)
---> 14         for item in train_generator.class_indices:   # Python's for loops are a "for each" loop
     15             if (result[0] == train_generator.class_indices[item]):
     16                 print(filename, item)

NameError: name 'train_generator' is not defined