Значение images.shape неприемлемо

Question

При использовании Nvlab StyleGan в dataset_tool.py, он показывает ошибку как:

value 'images.shape' is unsubscriptable

Вот код, показывающий ошибку:

def create_cifar10(tfrecord_dir, cifar10_dir):
print('Loading CIFAR-10 from "%s"' % cifar10_dir)
import pickle
images = []
labels = []
for batch in range(1, 6):
    with open(os.path.join(cifar10_dir, 'data_batch_%d' % batch), 'rb') as file:
        data = pickle.load(file, encoding='latin1')
    images.append(data['data'].reshape(-1, 3, 32, 32))
    labels.append(data['labels'])
images = np.concatenate(images)
labels = np.concatenate(labels)
assert images.shape == (50000, 3, 32, 32) and images.dtype == np.uint8
assert labels.shape == (50000,) and labels.dtype == np.int32
assert np.min(images) == 0 and np.max(images) == 255
assert np.min(labels) == 0 and np.max(labels) == 9
onehot = np.zeros((labels.size, np.max(labels) + 1), dtype=np.float32)
onehot[np.arange(labels.size), labels] = 1.0

with TFRecordExporter(tfrecord_dir, images.shape[0]) as tfr:
    order = tfr.choose_shuffled_order()
    for idx in range(order.size):
        tfr.add_image(images[order[idx]])
    tfr.add_labels(onehot[order])