rows = struct.unpack('>I', f.read(4))[0]
cols = struct.unpack('>I', f.read(4))[0]
images = np.frombuffer(f.read(), dtype=np.uint8, count = num * rows * cols)
- # split them up into an array of flat image vectors, so that first axis corresponds to labels
- images = images.reshape(num, rows * cols)
+ # split them up into an array of flat image column vectors
+ images = images.reshape(num, rows * cols).T
with open(labels_file, 'rb') as f:
# validate magic