mirror of
https://github.com/augustin64/projet-tipe
synced 2025-01-24 15:36:25 +01:00
33 lines
951 B
Python
33 lines
951 B
Python
|
#!/usr/bin/python3
|
||
|
|
IMAGES_MAGIC_NUMBER = 2051
|
||
|
|
LABELS_MAGIC_NUMBER = 2049
|
||
|
|
|
||
|
|
|
||
|
|
def write_images_to_binary(images, filepath):
|
||
|
|
byteorder = "big"
|
||
|
|
|
||
|
|
bytes_ = IMAGES_MAGIC_NUMBER.to_bytes(4, byteorder=byteorder)
|
||
|
|
bytes_ += (len(images)).to_bytes(4, byteorder=byteorder)
|
||
|
|
bytes_ += len(images[0][0]).to_bytes(4, byteorder=byteorder)
|
||
|
|
bytes_ += len(images[0]).to_bytes(4, byteorder=byteorder)
|
||
|
|
|
||
|
|
for image in images:
|
||
|
|
for row in image:
|
||
|
|
for nb in row:
|
||
|
|
bytes_ += int(nb).to_bytes(1, byteorder=byteorder)
|
||
|
|
|
||
|
|
with open(filepath, "wb") as f:
|
||
|
|
f.write(bytes_)
|
||
|
|
|
||
|
|
|
||
|
|
def write_labels_to_binary(labels, filepath):
|
||
|
|
byteorder = "big"
|
||
|
|
|
||
|
|
bytes_ = LABELS_MAGIC_NUMBER.to_bytes(4, byteorder=byteorder)
|
||
|
|
bytes_ += (len(labels)).to_bytes(4, byteorder=byteorder)
|
||
|
|
for label in labels:
|
||
|
|
bytes_ += int(label).to_bytes(1, byteorder=byteorder)
|
||
|
|
|
||
|
|
with open(filepath, "wb") as f:
|
||
|
|
f.write(bytes_)
|