import os

import json

import ftplib

import logging as log

thisDir=os.path.dirname(__file__)

with open(os.path.join(thisDir,"ftp.json")) as f:

	cfg=json.load(f)

def push(path):

	ftp.connect(cfg["host"],cfg["port"])

	ftp.login(cfg["user"],cfg["password"])

	filename=os.path.basename(path)

	log.info("uploading %s",path)

	with open(path,mode="rb") as f:

		ftp.storbinary("STOR "+filename,f)

	ftp.close()

	model.add(Dense(500,activation="relu"))

	model.add(Dropout(0.1))

	model.add(Dense(128,activation="relu"))

	model.add(Dropout(0.1))

	model.add(Dense(8))

	model.compile(optimizer='adam',loss='mse',metrics=['mae','accuracy'])

	return model

model=createCNN()

if args.load_model:

	model=load_model(args.load_model)

log.info("loading data...")

with np.load(args.data) as data:

	trainImages=data["trainImages"]

	trainLabels=data["trainLabels"]

	testImages=data["testImages"]

	testLabels=data["testLabels"]

log.info("done")

	model.fit(trainImages.reshape((-1,224,224,1)),trainLabels,epochs=(i+1)*10,initial_epoch=i*10,batch_size=128,validation_split=0.2)

	log.info("saving model...")

	model.save(path)

log.info(model.evaluate(testImages,testLabels))