from PIL import Image

import sys

filePath = ''
number = 0

if len(sys.argv) is 3:
filePath = sys.argv[1]
number = int(sys.argv[2])

elif len(sys.argv) is 2:
filePath = sys.argv[1]
number = int(input("number: "))

elif len(sys.argv) is 1:
filePath = input("image file path: ")
number = int(input("number: "))

else:
print("Usage:\n" + sys.argv[0] + "\n" + sys.argv[0] + " <image file path>")

data = b'\x00\x00\x08'

im = Image.open(filePath)
imWidth, imHeight = im.size
data = data + b'\x03\x00\x00\x00\x02'
data = data + imWidth.to_bytes(4, byteorder = 'big')
data = data + imHeight.to_bytes(4, byteorder = 'big')

pix = im.load()

for itX in range(0, imWidth):

for itY in range(0, imHeight):
data = data + b'\x00'

for itX in range(0, imWidth):

for itY in range(0, imHeight):
data = data + pix[itY, itX][0].to_bytes(1, byteorder = 'big')

with open('out.idx', 'wb') as f:

f.write(data)

data = b'\x00\x00\x08\x01\x00\x00\x00\x02\x09' + number.to_bytes(1, byteorder =

'big')
with open('labels.idx', 'wb') as f:
f.write(data)