def main(img_dir, img_name, classes):
config = Config()
config.batch_size = 1
config.num_classes = 2
result = {}
for index in classes:
result[index] = predict(img_dir + img_name, index, config)
print(result)
return result