seti10.py

from keras.preprocessing.image import img_to_array, load_img
import glob
import matplotlib.pyplot as plt
%matplotlib inline

data_signal_files = glob.glob(TRAIN_DIR+'/*/*.png')
sample_files = [data_signal_files[idx] for idx in range(0, 5600, 800+15)]
fix, ax = plt.subplots(2, 4, figsize=(14, 6))

for idx, img in enumerate(sample_files):
    id1 = 1 if idx > 3 else 0
    id2 = idx % 4
    img_arr = img_to_array(load_img(img))
    f = ax[id1, id2].imshow(img_arr / 255., aspect='auto')
    t = ax[id1, id2].set_title(img.split('/')[-1].split('.')[0].split('_')[-1], fontsize=10)
ax[1,3].set_axis_off()