def solve(bot, update):
file_id = update.message.photo[-1]
newFile = bot.getFile(file_id)
newFile.download('download.jpg')
bot.sendMessage(chat_id=update.message.chat_id, text="Processing...")
img = imread('download.jpg', IMREAD_GRAYSCALE)
json_file = open(args.model + '.json', 'r')
loaded_model_json = json_file.read()
json_file.close()
model = model_from_json(loaded_model_json)
model.load_weights(args.model + '.h5')
grid = get_grid(img, model, method='cnn')
B = Board(9, grid)
# print(B.grid)
solution_img(B.solution().T, B.grid.T, 'solution.png')
bot.send_photo(chat_id=update.message.chat_id,
photo=open('solution.png', 'rb'))
def save_generated_dataset(folder, number, randfont=True):
''' Save <number> of pictures in the given <folder>s orig subfolder and data as data.csv '''
y = np.zeros((number, 81), int)
for i in range(number):
picname = folder + 'orig/grid' + str(i).zfill(4) + '.png'
B = Board(9)
grid = B.solution()
fonts = [
'arial.ttf', 'arialbd.ttf', 'APHont-Regular_q15c.ttf',
'APHont-Bold_q15c.ttf', 'Hack.ttf', 'DejaVuSansMono-Bold.ttf',
'DroidSerif-Regular.ttf', 'OpenSans-Bold.ttf'
]
if randfont:
font = ImageFont.truetype('fonts/' + choice(fonts), size=45)
else:
font = ImageFont.truetype('fonts/APHont-Regular_q15c.ttf', size=45)
draw_matrix(grid, picname, frame=True, font=font)
y[i] = grid.reshape(81)
np.savetxt(folder + 'data.csv', y, delimiter=',', fmt='%d')
